integrating sqlalchemy

Author: bashwork

speleo.service/handlers/blockboard_handler.py

@@ -1,6 +1,6 @@
-import tornado.web
+import common
 
-class BlockboardHandler(tornado.web.RequestHandler):
+class BlockboardHandler(common.BaseHandler):
 
     RoutePath = r'/blockboard'
 

speleo.service/handlers/common.py

@@ -3,6 +3,14 @@
 
 class BaseHandler(tornado.web.RequestHandler):
 
+    @property
+    def db(self):
+        return self.application.database
+
+    @property
+    def cache(self):
+        return self.application.cache
+
     def get_current_user(self):
         user_json = self.get_secure_cooke("user")
         if not user_json: return None

speleo.service/handlers/contact_handler.py

@@ -1,6 +1,6 @@
-import tornado.web
+import common
 
-class ContactHandler(tornado.web.RequestHandler):
+class ContactHandler(common.BaseHandler):
 
     RoutePath = r'/contact'
 

speleo.service/handlers/dashboard_handler.py

@@ -1,6 +1,6 @@
-import tornado.web
+import common
 
-class DashboardHandler(tornado.web.RequestHandler):
+class DashboardHandler(common.BaseHandler):
 
     RoutePath = r'/dashboard'
 

speleo.service/handlers/root_handler.py

@@ -1,8 +1,8 @@
-import tornado.web
+import common
 
-class RootHandler(tornado.web.RequestHandler):
+class RootHandler(common.BaseHandler):
 
     RoutePath = r'/'
 
     def get(self):
-       self.render('index.html')
+        self.redirect('/about')

speleo.service/handlers/search_handler.py

@@ -1,6 +1,6 @@
-import tornado.web
+import common
 
-class SearchHandler(tornado.web.RequestHandler):
+class SearchHandler(common.BaseHandler):
 
     RoutePath = r'/search'
 

speleo.service/handlers/settings_handler.py

@@ -1,6 +1,6 @@
-import tornado.web
+import common
 
-class SettingsHandler(tornado.web.RequestHandler):
+class SettingsHandler(common.BaseHandler):
 
     RoutePath = r'/settings'
 

speleo.service/handlers/simple_handler.py

@@ -1,27 +1,27 @@
-import tornado.web
+import common
 
-class AboutHandler(tornado.web.RequestHandler):
+class AboutHandler(common.BaseHandler):
 
     RoutePath = r'/about'
 
     def get(self):
         self.render('about.html')
 
-class AnalyzeHandler(tornado.web.RequestHandler):
+class AnalyzeHandler(common.BaseHandler):
 
     RoutePath = r'/analyze'
 
     def get(self):
         self.render('analyze.html')
 
-class QueryHandler(tornado.web.RequestHandler):
+class QueryHandler(common.BaseHandler):
 
     RoutePath = r'/query'
 
     def get(self):
         self.render('query.html')
 
-class HealthHandler(tornado.web.RequestHandler):
+class HealthHandler(common.BaseHandler):
 
     RoutePath = r'/health'
 

speleo.service/main.py

@@ -3,6 +3,7 @@
 import inspect
 import logging
 import handlers
+import models
 import tornado.ioloop
 import tornado.web
 from tornado.options import define, options
@@ -13,38 +14,56 @@
 define("debug", default=False, help="Set to true to enable debugging", type=bool)
 define("port", default=8888, help="The port to run the service on", type=int)
 define('security', default='none', help='none|ldap', type=str)
+define('database', default='sqlite:////tmp/example.db', help="The database connection string to use", type=str)
 define('ldap_host', default='ldap://127.0.0.1', help='The ldap service to authenticate against', type=str)
 define('ldap_domain', default='', help='The domain to authenticate users under', type=str)
 define('ldap_basedn', default='', help='The base dn to search for users under', type=str)
 define('ldap_bind_dn', default=None, help='A priviledged user to perform ldap binds', type=str)
 define('ldap_bind_password', default=None, help='The priviledged user password', type=str)
 
 # ------------------------------------------------------------ 
-# settings
+# application
 # ------------------------------------------------------------ 
-settings = {
-    'static_path': os.path.join(os.path.dirname(__file__), 'static'),
-    'template_path': os.path.join(os.path.dirname(__file__), 'templates'),
-    'login_url':'/auth/login',
-    'cookie_secret':'something random should go here',
-    'xsrf_cookies':True,
-    'debug': options.debug,
-}
+class SpeleoApplication(tornado.web.Application):
 
-# ------------------------------------------------------------ 
-# handlers
-# ------------------------------------------------------------ 
-search = inspect.getmembers(handlers)
-routes = [(h.RoutePath, h) for n,h in search if 'Handler' in n]
+    def __init__(self):
+        # ---------------------------------------------------- 
+        # settings
+        # ---------------------------------------------------- 
+        root_path = os.path.dirname(__file__)
+        settings = {
+            'static_path': os.path.join(root_path, 'static'),
+            'template_path': os.path.join(root_path, 'templates'),
+            'login_url':'/auth/login',
+            'cookie_secret':'something random should go here',
+            'xsrf_cookies':True,
+            'debug': options.debug,
+        }
+
+        # ---------------------------------------------------- 
+        # handlers
+        # ---------------------------------------------------- 
+        search = inspect.getmembers(handlers)
+        routes = [(h.RoutePath, h) for n, h in search if 'Handler' in n]
+
+        # ---------------------------------------------------- 
+        # shared
+        # ---------------------------------------------------- 
+        self.database = models.get_database(options.database, options.debug)
+        #self.cache = memcache.Client([options.cache], debug=options.debug)
+
+        # ---------------------------------------------------- 
+        # initialize
+        # ---------------------------------------------------- 
+        tornado.web.Application.__init__(self, routes, **settings)
+        logging.debug('Installed Routes')
+        for route in routes: logging.debug(route)
 
 # ------------------------------------------------------------ 
-# main service
+# service
 # ------------------------------------------------------------ 
 if __name__ == "__main__":
     tornado.options.parse_config_file('settings.py')
     tornado.options.parse_command_line()
-    application = tornado.web.Application(routes, **settings)
-    logging.debug('Installed Handlers')
-    for route in routes: logging.debug(route)
-    application.listen(options.port)
+    SpeleoApplication().listen(options.port)
     tornado.ioloop.IOLoop.instance().start()

speleo.service/models.py

@@ -0,0 +1,37 @@
+from sqlalchemy import create_engine
+from sqlalchemy import Column, Integer, String, DateTime, Boolean
+from sqlalchemy.ext.declarative import declarative_base
+
+# ------------------------------------------------------------ 
+# utility methods
+# ------------------------------------------------------------ 
+Base = declarative_base()
+
+def get_database(connection, debug=False):
+    ''' Retrieve a database instance from the given config
+
+    :param connection: The connection string to the database
+    :param debug: Set to true to debug the operations
+    '''
+    engine = create_engine(connection, convert_unicode=True, echo=debug)
+    Base.metadata.create_all(engine)
+    models.init_db(engine)
+    return scoped_session(sessionmaker(bind=engine))
+
+
+# ------------------------------------------------------------ 
+# models
+# ------------------------------------------------------------ 
+class User(Base):
+
+    __tablename__ = 'users'
+
+    id = Column(Integer, primary_key=True)
+    username = Column(String(30), nullable=False)
+    first_name = Column(String(30), nullable=False)
+    last_name = Column(String(30), nullable=False)
+    email = Column(String(75), nullable=False)
+    password = Column(String(128), nullable=False)
+
+    def __repr__(self):
+        return "<User(%s)>" % (self.username)