Schema

from passwordsafe.models import Base
from sqlalchemy.orm import relationship


from sqlalchemy import (\
        Column,
        Integer,
        String
    )

class Data(Base):

    __tablename__ = 'data'

    id = Column('data_id', Integer, primary_key=True)
    entry = Column('data_entry', String(256), nullable=False)
    public_key = Column('data_public_key', String(1024), nullable=False)

    def __init__(self, entry, public_key):
        self.entry = entry
        self.public_key = public_key

    def __repr__(self):
        return "<Data(entry='%s', public_key='%s')>" % (
                    self.entry, self.public_key)

class User(Base):

    __tablename__ = 'users'

    id = Column('user_id', Integer, primary_key=True)
    username = Column('user_name', String(20), nullable=False)
    password = Column('user_pass', String(512), nullable=False)
    email = Column('user_email', String(256), nullable=False)
    salt = Column('user_salt', String(256), nullable=False)
    public_key = Column('user_public_key', String(1024), nullable=False)
    private_key = Column('user_private_key', String(1024), nullable=False)
    data_entries = relationship("UserData", backref="user")

    def __repr__(self):
        return "<User(username='%s', password='%s', email='%s', salt='%s', public_key='%s', private_key='%s')>" % (
                    self.username, self.password, self.email, self.salt, self.public_key, self.private_key)

    def __init__(self, username, password, email, salt, public_key, private_key):
        self.username = username
        self.password = password
        self.email = email
        self.salt = salt
        self.public_key = public_key
        self.private_key = private_key


class UserData(Base):

    __tablename__ = 'user_data_map'

    user_id = Column('user_id', Integer, ForeignKey('users.user_id'), primary_key=True)
    data_id = Column('data_id', Integer, ForeignKey('data.data_id'), primary_key=True)
    data_private_key = Column('data_private_key', String(1024), nullable=False)
    data = relationship("Data", backref="userdata")