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()