mirror of
https://github.com/PaiGramTeam/sqlmodel.git
synced 2024-11-25 17:35:40 +00:00
31 lines
696 B
Python
31 lines
696 B
Python
from sqlmodel import Session
|
||
|
||
from .database import create_db_and_tables, engine
|
||
from .hero_model import Hero
|
||
from .team_model import Team
|
||
|
||
|
||
def create_heroes():
|
||
with Session(engine) as session:
|
||
team_z_force = Team(name="Z-Force", headquarters="Sister Margaret’s Bar")
|
||
|
||
hero_deadpond = Hero(
|
||
name="Deadpond", secret_name="Dive Wilson", team=team_z_force
|
||
)
|
||
session.add(hero_deadpond)
|
||
session.commit()
|
||
|
||
session.refresh(hero_deadpond)
|
||
|
||
print("Created hero:", hero_deadpond)
|
||
print("Hero's team:", hero_deadpond.team)
|
||
|
||
|
||
def main():
|
||
create_db_and_tables()
|
||
create_heroes()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|