models.py

from peewee import *
from datetime import datetime
from config import DATABASE
from playhouse.migrate import *

pg_db = PostgresqlDatabase(DATABASE['NAME'], user=DATABASE['USER'], password=DATABASE['PASSWORD'], host=DATABASE['HOST'], port=DATABASE['PORT'], autorollback=True)
#migrator = PostgresqlMigrator(pg_db)

class User(Model):
    username = CharField(max_length=64, unique=True)
    email = CharField(max_length=128, unique=True)
    password = CharField(max_length=255, default='')
    fullname = CharField(max_length=128, null=True)
    location = CharField(max_length=128, null=True)
    company = CharField(max_length = 128, null=True)
    biography = TextField(null=True)
    website = TextField(null=True)
    birthdate = DateField(null=True)
    email_public = BooleanField(default=False)
    display_birthday = BooleanField(default=False)
    hide_inappropriate = BooleanField(default=True)
    enabled = BooleanField(default=False)
    type = CharField(max_length=11, default="user") #EnumField('administrator', 'moderator', 'user')
    remote_address = CharField(max_length=39, null=True)
    deleted_at = DateTimeField(null=True)
    created_at = DateTimeField(default=datetime.now)
    updated_at = DateTimeField(default=datetime.now)
    class Meta:
        database = pg_db
        db_table = "users"
    def __str__(self):
        return self.username

class Item(Model):
    user = ForeignKeyField(User)
    revision_id = IntegerField(null=True)
    upload_session_id = IntegerField(null=True)
    name = CharField(max_length=255)
    description = TextField(null=True)
    instructions = TextField(null=True)
    tags = TextField(null=True)
    license = CharField(max_length=10) #EnumField
    version = CharField(max_length=50)
    flagged_at = DateTimeField(null=True)
    deleted_at = DateTimeField(null=True)
    created_at = DateTimeField(default=datetime.now)
    updated_at = DateTimeField(default=datetime.now)
    item_likes = IntegerField(default=0)
    item_dislikes = IntegerField(default=0)
    item_views = IntegerField(default=0)
    
    class Meta:
        database = pg_db
        db_table = "items"
    def __str__(self):
            return self.name

class ItemRevision(Model):
    item = ForeignKeyField(Item, null=True, backref="item_revisions")
    user = ForeignKeyField(User)
    name = CharField(max_length=255, default="")
    description = TextField(null=True, default="")
    instructions = TextField(null=True, default="")
    tags = TextField(null=True, default="")
    license = CharField(max_length=10, default="") #Enum
    version = CharField(max_length=50, default="")
    upload_session_id = IntegerField(null=True)
    deleted_at = DateTimeField(null=True)
    created_at = DateTimeField(default=datetime.now)
    updated_at = DateTimeField(default=datetime.now)
    @classmethod
    def disable_relationships(self):
        ItemRevision.item = IntegerField()
        ItemRevision.user = IntegerField()
        return ItemRevision
    @classmethod
    def enable_relationships(self):
        item = ForeignKeyField(Item, null=True, backref="item_revisions")
        user = ForeignKeyField(User)
        return ItemRevision
    class Meta:
        database = pg_db
        db_table = "item_revisions"
    def __str__(self):
        return self.name


class ItemImage(Model):
    revision = ForeignKeyField(ItemRevision, null=True, backref="item_images")
    path = CharField(max_length=255, default ="")
    upload_id = IntegerField()
    width = IntegerField()
    height = IntegerField()
    type = CharField(max_length=64, default ="")
    @classmethod
    def disable_relationships(self):
        ItemImage.revision = IntegerField()
        return ItemImage
    @classmethod
    def enable_relationships(self):
        revision = ForeignKeyField(ItemRevision, null=True, backref="item_images")
        return ItemImage
    class Meta:
        database = pg_db
        db_table = "item_images"
    def __str__(self):
        return self.type



class ItemLike(Model):
    item = ForeignKeyField(Item)
    user = ForeignKeyField(User)
    score = IntegerField()
    remote_address = CharField(max_length = 64)

    class Meta:
        database = pg_db
        db_table = "item_likes"
    def __str__(self):
        return self.type


class ItemUpload(Model):
    uploader = ForeignKeyField(User, null=True)
    file_location = CharField(max_length=128)
    filename = CharField(max_length=256)
    file_extension = CharField(max_length=16, null=True) #adding the nullable for testing
    size = IntegerField()#filesize Number (file size in number of bytes)
    mimetype = CharField(max_length=64)#mimetype  varchar(64) NULLABLE (ex: image/png)
    upload_session_id = IntegerField(null=True)
    revision = ForeignKeyField(ItemRevision, null=True)
	#type varchar(1) NULLABLE (either "primary", "secondary") //define it but mostly ignore for now
	#hash String(40) NULLABLE (file hash, md5 ) //define but can ignore for now
	#enabled Boolean
	#deleted Boolean
	#date_uploaded Timestamp
    created_at = DateTimeField(default=datetime.now)
    updated_at = DateTimeField(default=datetime.now)
	#remote_address varchar(39)id #adding the nullable for testing
    @classmethod
    def disable_relationships(self):
        ItemUpload.uploader = IntegerField()
        ItemUpload.revision = IntegerField()
        return ItemUpload
    @classmethod
    def enable_relationships(self):
        uploader = ForeignKeyField(User, null=True)
        revision = ForeignKeyField(ItemRevision, null=True)
        return ItemUpload
    class Meta:
        database = pg_db
        db_table = "item_uploads"
    def __str__(self):
        return self.filename

class ItemView(Model):
    item = ForeignKeyField(Item)
    revision = ForeignKeyField(ItemRevision, null=True)
    user = ForeignKeyField(User, null = True)
    timestamp =  DateTimeField(default=datetime.now)
    remote_address = CharField(max_length = 64)
    class Meta:
        database = pg_db
        db_table = "item_views" 

class File(Model):
    item = ForeignKeyField(Item, null = True)
    revision = ForeignKeyField(ItemRevision, null = True)
    name = CharField(max_length = 255)
    downloadable = BooleanField(default=False)
    sort_order = IntegerField()
    render = BooleanField(default=True)
    preview = BooleanField(default=True)

    class Meta:
            database = pg_db
            db_table = "files"


class Download(Model):
    item = ForeignKeyField(Item, null=True)
    revision = ForeignKeyField(ItemRevision, null=True)
    file = ForeignKeyField(File, null=True)
    user = ForeignKeyField(User, null=True)
    created_at = DateTimeField(default=datetime.now)
    class Meta:
        database = pg_db
        db_table = "downloads"


class PasswordReset(Model):
    user = ForeignKeyField(User)
    reset_key = CharField(max_length=32)
    created_at = DateTimeField(default=datetime.now)
    class Meta:
        database = pg_db
        db_table = "reset_passwords"