from models import UserAccount, Complex
from peewee import (PostgresqlDatabase)
def create_users():
UserAccount.create(name='Tony')
UserAccount.create(name='Bruce')
def create_complex():
Complex.create(name='Prasads')
Complex.create(name='PVR')
pg_db = PostgresqlDatabase('movies',
user='******',
password='******',
host='0.0.0.0',
port=5444)
pg_db.connect()
create_users()
create_complex()
pg_db.commit()
class Meta:
table_name = 'movies_category'
class MoviesCompany(BaseModel):
company_id = ForeignKeyField(Companies)
movie_id = ForeignKeyField(Movies)
class Meta:
table_name = 'movies_company'
class Reviews(BaseModel):
review = TextField()
score = IntegerField()
user_id = ForeignKeyField(Users)
movie_id = ForeignKeyField(Movies)
class Meta:
table_name = 'reviews'
TABLES = [
Categories, Companies, Movies, Users, MoviesCategory, Reviews,
MoviesCompany
]
with database.connection_context():
database.create_tables(TABLES, safe=True)
database.commit()
