from peewee import *
from datetime import datetime

db = SqliteDatabase('test.db')


class User(Model):
    first_name = CharField()
    last_name = CharField()
    username = CharField()
    password = CharField()


class Photo(Model):
    name = CharField()
    user = ForeignKeyField(User, related_name="photos")
    meta_data = CharField()

    class Meta:
        database = db


class Category(Model):
    category = CharField()
    photo = ForeignKeyField(Photo, related_name='categories')



if __name__ == '__main__':

    db.connect()
    db.create_tables(models=[Photo, Category, User], safe=True)

    for i in range(1,10):
        u = User(first_name='first {}'.format(i),last_name="last {}".format(i),
                 username="u{}".format(i),password='pass{}'.format(i))
        p = Photo(user=u, name='Photo {}'.format(str(i)), meta_data="meta data {}".format(i))
        c = Category(category="category {}".format(str(i)), photo=p)
        u.save()
        p.save()
        c.save()

    print([(i.name, i.meta_data) for i in Photo.select()])
    print([(i.category, i.photo.name) for i in Category.select()])
    print([[x.name for x in i.photos] for i in User.select()])