1 files changed, 86 insertions, 0 deletions
diff --git a/module/web/utils.py b/module/web/utils.py
new file mode 100644
index 000000000..cf3f2d5f3
--- /dev/null
+++ b/module/web/utils.py
@@ -0,0 +1,86 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 3 of the License,
+    or (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+    See the GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this plrogram; if not, see <http://www.gnu.org/licenses/>.
+
+    @author: RaNaN
+"""
+from os.path import join, abspath, commonprefix
+
+from bottle import request, HTTPError, redirect, ServerAdapter
+
+from webinterface import env, TEMPLATE
+
+def render_to_response(name, args={}, proc=[]):
+    for p in proc:
+        args.update(p())
+
+    t = env.get_template(join(TEMPLATE, name))
+    return t.render(**args)
+
+def parse_permissions(session):
+    perms = {"can_change_status": False,
+            "can_see_dl": False}
+
+    if not session.get("authenticated", False):
+        return perms
+
+    perms["can_change_status"] = True
+    perms["can_see_dl"] = True
+
+    return perms
+
+def parse_userdata(session):
+    return {"name": session.get("name", "Anonymous"),
+            "is_staff": True,
+            "is_authenticated": session.get("authenticated", False)}
+
+def formatSize(size):
+    """formats size of bytes"""
+    size = int(size)
+    steps = 0
+    sizes = ["KB", "MB", "GB", "TB"]
+
+    while size > 1000:
+        size /= 1024.0
+        steps += 1
+
+    return "%.2f %s" % (size, sizes[steps])
+
+def login_required(perm=None):
+    def _dec(func):
+        def _view(*args, **kwargs):
+            s = request.environ.get('beaker.session')
+            if s.get("name", None) and s.get("authenticated", False):
+                if perm:
+                    pass
+                    #print perm
+                return func(*args, **kwargs)
+            else:
+                if request.header.get('X-Requested-With') == 'XMLHttpRequest':
+                    return HTTPError(403, "Forbidden")
+                else:
+                    return redirect("/login")
+
+        return _view
+
+    return _dec
+
+class CherryPyWSGI(ServerAdapter):
+
+    def run(self, handler):
+        from wsgiserver import CherryPyWSGIServer
+
+        server = CherryPyWSGIServer((self.host, self.port), handler)
+        server.start()