Webservice side projects

webservice/bam/application/modules/admin/__init__.py

@@ -0,0 +1,115 @@
+from application import app, db
+#from application import thumb
+
+from flask import render_template, redirect, url_for
+from flask.ext.sqlalchemy import SQLAlchemy
+from flask.ext import admin, login
+from flask.ext.admin import Admin, expose
+from flask.ext.admin import form
+from flask.ext.admin.contrib import sqla
+from flask.ext.admin.contrib.sqla import ModelView
+from flask.ext.admin.base import BaseView
+from flask.ext.security import current_user
+
+from werkzeug import secure_filename
+from jinja2 import Markup
+from wtforms import fields, validators, widgets
+from wtforms.fields import SelectField, TextField
+import os, hashlib, time
+import os.path as op
+
+
+def _list_items(view, context, model, name):
+    """Utilities to upload and present images
+    """
+    if not model.name:
+        return ''
+    return Markup(
+        '<div class="select2-container-multi">'
+            '<ul class="select2-choices" style="border:0;cursor:default;background:none;">%s</ul></div>' % (
+                ''.join( ['<li class="select2-search-choice" style="padding:3px 5px;">'
+                            '<div>'+item.name+'</div></li>' for item in getattr(model,name)] )))
+
+
+def _list_thumbnail(view, context, model, name):
+    if not getattr(model,name):  #model.name only does not work because name is a string
+        return ''
+    return ''
+    # return Markup('<img src="%s">' % url_for('static', 
+    #     filename=thumb.thumbnail(getattr(model,name), '50x50', crop='fit')))
+
+# Create directory for file fields to use
+file_path = op.join(op.dirname(__file__), '../../static/files',)
+try:
+    os.mkdir(file_path)
+except OSError:
+    pass
+
+
+def prefix_name(obj, file_data):
+    # Collect name and extension
+    parts = op.splitext(file_data.filename)
+    # Get current time (for unique hash)
+    timestamp = str(round(time.time()))
+    # Has filename only (not extension)
+    file_name = secure_filename(timestamp + '%s' % parts[0])
+    # Put them together
+    full_name = hashlib.md5(file_name).hexdigest() + parts[1]
+    return full_name
+
+
+def image_upload_field(label):
+    return form.ImageUploadField(label,
+                    base_path=file_path,
+                    thumbnail_size=(100, 100, True),
+                    namegen=prefix_name,
+                    endpoint='filemanager.static')
+
+
+# Define wtforms widget and field
+class CKTextAreaWidget(widgets.TextArea):
+    def __call__(self, field, **kwargs):
+        kwargs.setdefault('class_', 'ckeditor')
+        return super(CKTextAreaWidget, self).__call__(field, **kwargs)
+
+
+class CKTextAreaField(fields.TextAreaField):
+    widget = CKTextAreaWidget()
+
+
+# Create customized views with access restriction
+class CustomModelView(ModelView):
+    def is_accessible(self):
+        return True
+        #return login.current_user.has_role('admin')
+
+class CustomBaseView(BaseView):
+    def is_accessible(self):
+        return True
+        #return login.current_user.has_role('admin')
+
+
+# Create customized index view class that handles login & registration
+class CustomAdminIndexView(admin.AdminIndexView):
+    def is_accessible(self):
+        return True
+        #return login.current_user.has_role('admin')
+
+    @expose('/')
+    def index(self):
+        return super(CustomAdminIndexView, self).index()
+
+    @expose('/logout/')
+    def logout_view(self):
+        login.logout_user()
+        return redirect(url_for('homepage'))
+
+
+# Create admin
+backend = Admin(
+    app, 
+    'BAM', 
+    index_view=CustomAdminIndexView(), 
+    base_template='admin/layout_admin.html'
+)
+