adding sqla stuff

Author: timkpaine

tkp_utils/sqla/__init__.py

@@ -0,0 +1,42 @@
+import sys
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from .models import Base, User, APIKey
+
+
+def main(sql_url, application_name="TKP_UTILS", user_class=User, apikey_class=APIKey):
+    '''Create dummy notebook data for sqlalchemy'''
+    engine = create_engine(sql_url, echo=False)
+    Base.metadata.create_all(engine)
+    sm = sessionmaker(bind=engine)
+
+    session = sm()
+    admin = User(username='admin',
+                 password='admin',
+                 email='test@test.com',
+                 admin=True)
+    try:
+        session.add(admin)
+        session.commit()
+        session.refresh(admin)
+        print('added admin: {}'.format(admin))
+        key = APIKey(user=admin)
+        session.add(key)
+        session.commit()
+        session.refresh(key)
+        print('added apikey: {}'.format(key))
+        with open("keys.sh", "w") as fp:
+            fp.write("#!/bin/bash\n")
+            fp.write("export TKP_UTILS={}\n".format(key.key))
+            fp.write("export TKP_UTILS={}\n".format(key.secret))
+    except BaseException:
+        session.rollback()
+        admin = session.query(User).filter_by(username='test').first()
+        print('admin exists: {}'.format(admin))
+
+
+if __name__ == '__main__':
+    if len(sys.argv) < 2:
+        print("args: <sql_url>")
+    else:
+        main(sys.argv[1])

tkp_utils/sqla/fixtures.py

@@ -0,0 +1,72 @@
+import pandas as pd
+import six
+import secrets
+import ujson
+import validators
+from datetime import datetime, timedelta
+from sqlalchemy import Column, Integer, String, Boolean, DateTime, JSON, ForeignKey
+from sqlalchemy.ext.hybrid import hybrid_property
+from sqlalchemy.orm import relationship
+from sqlalchemy.ext.declarative import declarative_base
+
+TOKEN_WIDTH = 64
+Base = declarative_base()
+
+
+class User(Base):
+    __tablename__ = 'users'
+    id = Column(Integer, primary_key=True)
+    username = Column(String(100), nullable=False, unique=True)
+    password = Column(String(100), nullable=False)
+
+    _email = Column("email", String, nullable=False, unique=True)
+
+    apikeys = relationship('APIKey', back_populates='client')
+    admin = Column(Boolean, default=False)
+
+    @hybrid_property
+    def email(self):
+        return self._email
+
+    @email.setter
+    def email(self, email):
+        # TODO validate
+        self._email = email
+
+    def __repr__(self):
+        return "<User(id='{}', username='{}')>".format(self.id, self.username)
+
+    def to_dict(self):
+        ret = {}
+        for item in ("id", "username", "email"):
+            ret[item] = getattr(self, item)
+        return ret
+
+    def from_dict(self, d):
+        raise NotImplementedError()
+
+
+class APIKey(Base):
+    __tablename__ = 'apikeys'
+    id = Column(Integer, primary_key=True)
+    user_id = Column(Integer, ForeignKey('users.id', ondelete='cascade'))
+    user = relationship('User', back_populates='apikeys')
+    key = Column(String(100), nullable=False, default=lambda: secrets.token_urlsafe(TOKEN_WIDTH))
+    secret = Column(String(100), nullable=False, default=lambda: secrets.token_urlsafe(TOKEN_WIDTH))
+
+    @staticmethod
+    def generateKey():
+        return {'key': secrets.token_urlsafe(TOKEN_WIDTH),
+                'secret': secrets.token_urlsafe(TOKEN_WIDTH)}
+
+    def __repr__(self):
+        return "<Key(id='{}', key='{}', secret='***')>".format(self.id, self.key)
+
+    def to_dict(self):
+        ret = {}
+        for item in ("id", "user_id", "key", "secret"):
+            ret[item] = getattr(self, item)
+        return ret
+
+    def from_dict(self, d):
+        raise NotImplementedError()

tkp_utils/sqla/models.py

@@ -5,6 +5,7 @@
 from concurrent.futures import ThreadPoolExecutor
 from contextlib import contextmanager
 from jinja2 import Environment, FileSystemLoader, TemplateNotFound
+from ...sqla.models import User, APIKey
 from ...web import parse_body
 
 
@@ -145,8 +146,8 @@ def render_template(self, template, **kwargs):
 
     def initialize(self,
                    sessionmaker=None,
-                   UserSQLClass=None,
-                   APIKeySQLClass=None,
+                   UserSQLClass=User,
+                   APIKeySQLClass=APIKey,
                    user_id_field='id',
                    apikey_id_field='id',
                    user_apikeys_field='apikeys',