示例#1
0
 def create_db():
     """Create DB."""
     # from pmaapi.api.open_model.open_model_py.__main__ import db
     from flask_sqlalchemy import SQLAlchemy, declarative_base
     from pmaapi.api.open_model_py.__main__ import OpenModel
     from pmaapi.config import MODEL_FILE
     db = SQLAlchemy(FLASK_APP)
     db.Base = declarative_base()
     # TODO: Figure out how to get 'db' to know about this,
     #   if it doesn't automatically (reading from globals?).
     # models = OpenModel(MODEL_FILE)
     db.create_all()  # Run if it does not exist.
     db.session.commit()
示例#2
0
# -*- coding: utf-8 -*-
"""Temporary SqlAlchemy Models."""
from sys import stderr
from copy import copy
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, DateTime, Boolean, Integer, String, Float, \
    BigInteger, Date
from sqlalchemy.sql.functions import func as sqlalchemy_func
from flask_sqlalchemy import SQLAlchemy
from pmaapi.__main__ import FLASK_APP
from pmaapi.definitions.error import DatabaseException
# from sqlalchemy.exc import ProgrammingError, IntegrityError
# from pmaapi.definitions.error import raise_database_exception

db = SQLAlchemy(FLASK_APP)
db.Base = declarative_base()
now = sqlalchemy_func.now


# --- Testing and Placeholder Functions --- #
def get_current_user():
    """Get current user. Placeholder. Shouldn't even come from this class.

    Returns:
        str: Current user.
        """
    return 'System'  # Placeholder


# - TODO: Relational mapping - http://tinyurl.com/yc2j7jkg
# - TODO: Use unicode instead of string?