Initial YakPanel commit

class_v2/flask_compress_v2.py

@@ -0,0 +1,146 @@
+import sys,os
+from gzip import GzipFile
+from io import BytesIO
+
+from flask import request, current_app,session,Response,g,abort
+
+
+if sys.version_info[:2] == (2, 6):
+    class GzipFile(GzipFile):
+        """ Backport of context manager support for python 2.6"""
+        def __enter__(self):
+            if self.fileobj is None:
+                raise ValueError("I/O operation on closed GzipFile object")
+            return self
+
+        def __exit__(self, *args):
+            self.close()
+
+
+class DictCache(object):
+
+    def __init__(self):
+        self.data = {}
+
+    def get(self, key):
+        return self.data.get(key)
+
+    def set(self, key, value):
+        self.data[key] = value
+
+
+class Compress(object):
+    """
+    The Compress object allows your application to use Flask-Compress.
+
+    When initialising a Compress object you may optionally provide your
+    :class:`flask.Flask` application object if it is ready. Otherwise,
+    you may provide it later by using the :meth:`init_app` method.
+
+    :param app: optional :class:`flask.Flask` application object
+    :type app: :class:`flask.Flask` or None
+    """
+    def __init__(self, app=None):
+        """
+        An alternative way to pass your :class:`flask.Flask` application
+        object to Flask-Compress. :meth:`init_app` also takes care of some
+        default `settings`_.
+
+        :param app: the :class:`flask.Flask` application object.
+        """
+        self.app = app
+        if app is not None:
+            self.init_app(app)
+
+    def init_app(self, app):
+        defaults = [
+            ('COMPRESS_MIMETYPES', ['text/html', 'text/css', 'text/xml',
+                                    'application/json',
+                                    'application/javascript']),
+            ('COMPRESS_LEVEL', 9),
+            ('COMPRESS_MIN_SIZE', 500),
+            ('COMPRESS_CACHE_KEY', None),
+            ('COMPRESS_CACHE_BACKEND', None),
+            ('COMPRESS_REGISTER', True),
+        ]
+
+        for k, v in defaults:
+            app.config.setdefault(k, v)
+
+        backend = app.config['COMPRESS_CACHE_BACKEND']
+        self.cache = backend() if backend else None
+        self.cache_key = app.config['COMPRESS_CACHE_KEY']
+
+        if (app.config['COMPRESS_REGISTER'] and
+                app.config['COMPRESS_MIMETYPES']):
+            app.after_request(self.after_request)
+
+    def after_request(self, response):
+        app = self.app or current_app
+        accept_encoding = request.headers.get('Accept-Encoding', '')
+        response.headers['Server'] = 'nginx'
+        response.headers['Connection'] = 'keep-alive'
+        if not 'tmp_login' in session:
+            response.headers["X-Frame-Options"] = "SAMEORIGIN"
+        if 'dologin' in g and app.config['SSL']:
+            try:
+                for k,v in request.cookies.items():
+                    response.set_cookie(k,'',expires='Thu, 01-Jan-1970 00:00:00 GMT',path='/')
+            except:
+                pass
+
+        if 'rm_ssl' in g:
+            import public
+            try:
+                for k,v in request.cookies.items():
+                    response.set_cookie(k,'',expires='Thu, 01-Jan-1970 00:00:00 GMT',path='/')
+            except:
+                pass
+            session_name = app.config['SESSION_COOKIE_NAME']
+            session_id = public.get_session_id()
+            response.set_cookie(session_name,'',expires='Thu, 01-Jan-1970 00:00:00 GMT',path='/')
+            response.set_cookie(session_name, session_id, path='/', max_age=86400 * 30,httponly=True)
+
+            request_token = request.cookies.get('request_token','')
+            if request_token:
+                response.set_cookie('request_token',request_token,path='/',max_age=86400 * 30)
+
+        if (response.mimetype not in app.config['COMPRESS_MIMETYPES'] or
+            'gzip' not in accept_encoding.lower() or
+            not 200 <= response.status_code < 300 or
+            (response.content_length is not None and
+             response.content_length < app.config['COMPRESS_MIN_SIZE']) or
+            'Content-Encoding' in response.headers):
+            g.response = response
+            return response
+
+        response.direct_passthrough = False
+        if self.cache:
+            key = self.cache_key(response)
+            gzip_content = self.cache.get(key) or self.compress(app, response)
+            self.cache.set(key, gzip_content)
+        else:
+            gzip_content = self.compress(app, response)
+
+        response.set_data(gzip_content)
+
+        response.headers['Content-Encoding'] = 'gzip'
+        response.headers['Content-Length'] = response.content_length
+
+        vary = response.headers.get('Vary')
+        if vary:
+            if 'accept-encoding' not in vary.lower():
+                response.headers['Vary'] = '{}, Accept-Encoding'.format(vary)
+        else:
+            response.headers['Vary'] = 'Accept-Encoding'
+
+        g.response = response
+        return response
+
+    def compress(self, app, response):
+        gzip_buffer = BytesIO()
+        with GzipFile(mode='wb',
+                      compresslevel=app.config['COMPRESS_LEVEL'],
+                      fileobj=gzip_buffer) as gzip_file:
+            gzip_file.write(response.get_data())
+        return gzip_buffer.getvalue()