database_setup.py

from sqlalchemy import Column, ForeignKey, Integer, Boolean, String, DateTime, Date
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy import types

import datetime
from datetime import date
import time

Base = declarative_base()

RESTuri = 'http://localhost:5000/ondeck/api/v1.0/'

class EndUser(Base):
    __tablename__ = 'enduser'

    id = Column(String(250), primary_key=True)
    name = Column(String(250), nullable=False)
    pw_hash = Column(String(250), nullable=False)
    vision = Column(Integer, nullable=False)  # On Deck setting
    date_created = Column(DateTime, default=datetime.datetime.utcnow)

    @property
    def serialize(self):
        return {
            'id': self.id,
            'name': self.name,
            'vision': self.vision,
            'userURI': RESTuri + 'user/' + self.id,
            'createTaskURI': RESTuri + 'tasks/new/' + self.id,
            'doneTasksURI': RESTuri + 'tasks/' + self.id + '/done',
            'onDeckTasksURI': RESTuri + 'tasks/' + self.id + '/on-deck',
            'activeTasksURI': RESTuri + 'tasks/' + self.id + '/active'
        }


class Task(Base):
    __tablename__ = 'task'

    id = Column(String(250), primary_key=True)
    name = Column(String(250), nullable=False)
    task_group = Column(String(250), nullable=False)
    due_date = Column(Date, nullable=False)
    heads_up = Column(Date, nullable=True)
    done = Column(Boolean, nullable=False)
    completion_date = Column(Date, nullable=True)
    notes = Column(String(400), nullable=True)
    enduser_id = Column(String(250), ForeignKey('enduser.id'), nullable=False)
    enduser = relationship(EndUser)
    date_created = Column(DateTime, default=datetime.datetime.utcnow)

    @property
    def serialize(self):
        due = self.due_date
        now = datetime.date.today()
        daysLeft = (due-now).days

        return {
            'name': self.name,
            'task_group': self.task_group,
            'completion_date': str(self.completion_date) if self.completion_date else None,
            'due_date': str(self.due_date),
            'days_left': daysLeft,
            'notes': self.notes,
            'heads_up': str(self.heads_up) if self.heads_up else None,
            'done': self.done,
            'uri': RESTuri + 'tasks/' + self.id
        }


engine = create_engine('postgresql://me:password@localhost/tasks')
Base.metadata.create_all(engine)