summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xfbin.py31
1 files changed, 29 insertions, 2 deletions
diff --git a/fbin.py b/fbin.py
index 2bda843..9c8d31d 100755
--- a/fbin.py
+++ b/fbin.py
@@ -1,7 +1,7 @@
#!/usr/bin/env python2
import templates
-import settings, db, os, random, datetime, mimetypes, cgi, tempfile, hashlib, Cookie, urllib, subprocess
+import settings, db, os, random, datetime, mimetypes, cgi, tempfile, hashlib, Cookie, urllib, subprocess, json
from PIL import Image
base62_alphabet = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'
@@ -527,6 +527,32 @@ class Application(object):
'filename': file.filename,
}))
+ def api(self, environ, start_response, path):
+ c = Cookie.SimpleCookie(environ['HTTP_COOKIE'] if 'HTTP_COOKIE' in environ else None)
+ user = self.validate_cookie(c)
+ if user == None:
+ start_response('200 OK', [('Content-Type', 'application/json')])
+ return [json.dumps({'status': False, 'message': 'Not logged in'})]
+ form = cgi.FieldStorage(environ = environ)
+ method = form.getvalue('method')
+ data = {'status': False, 'method': method, 'message': None}
+ if method in ('list', 'images'):
+ files = self.get_files(user)
+ data['files'] = [
+ {
+ 'name': f.filename,
+ 'hash': f.hash,
+ 'date': int(f.date.strftime('%s')),
+ 'size': f.get_size(),
+ }
+ for f in files if method == 'list' or (method == 'images' and f.is_image())
+ ]
+ data['status'] = True
+ else:
+ data['message'] = 'Unknown method "%s"'
+ start_response('200 OK', [('Content-Type', 'application/json')])
+ return [json.dumps(data)]
+
f = file
u = upload
l = login
@@ -539,11 +565,12 @@ class Application(object):
r = register
c = changepass
d = delete
+ a = api
def __call__(self, environ, start_response):
path = environ['PATH_INFO'].split('/')[1:]
module = path[0] if len(path) else ''
- if len(module) and module in 'fulshmitorcd':
+ if len(module) and module in 'fulshmitorcda':
return getattr(self, module)(environ, start_response, path)
else:
start_response('302 Found', [('Location', settings.virtual_root + 'u')])