from sqlalchemy import create_engine, Table, Column, Text, String, Integer, DateTime, JSON, ForeignKey from sqlalchemy.orm import relationship import sqlalchemy.sql.functions as func from sqlalchemy_utc import UtcDateTime, utcnow from wtforms_alchemy import ModelForm from app.sql.common import db, Base, Session # from app.sql.models.page import Page from app.settings import app_cfg from os.path import join class Graph(Base): """Table for storing references to graphs""" __tablename__ = 'graph' id = Column(Integer, primary_key=True) path = Column(String(64, convert_unicode=True), nullable=False) title = Column(String(64, convert_unicode=True), nullable=False) description = Column(Text(convert_unicode=True), nullable=False) settings = Column(JSON, default={}, nullable=True) created_at = Column(UtcDateTime(), default=utcnow()) updated_at = Column(UtcDateTime(), onupdate=utcnow()) # pages = relationship('Page', lazy='dynamic') def toJSON(self): return { 'id': self.id, 'path': self.path, 'title': self.title, 'description': self.description, 'settings': self.settings, 'created_at': self.created_at, 'updated_at': self.updated_at, } def toFullJSON(self): data = self.toJSON() data['pages'] = [ page.toJSON() for page in self.pages ] return data class GraphForm(ModelForm): class Meta: model = Graph exclude = ['settings', 'created_at', 'updated_at'] def get_session(): return Session()