FilesExpand file tree

 main

/

setup_db.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

84 lines (59 loc) · 2.19 KB

 main

/

setup_db.py

Copy path

Top

File metadata and controls

• Code
• Blame

84 lines (59 loc) · 2.19 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

import os

from sqlalchemy import create_engine, Column, Integer, String, ForeignKey, Float

from sqlalchemy.ext.declarative import declarative_base

from sqlalchemy.orm import sessionmaker, relationship

Base = declarative_base()

class User(Base):

__tablename__ = "users"

id = Column(Integer, primary_key=True, index=True)

name = Column(String, index=True)

age = Column(Integer)

email = Column(String, unique=True, index=True)

orders = relationship("Order", back_populates="user")

class Food(Base):

__tablename__ = "food"

id = Column(Integer, primary_key=True, index=True)

name = Column(String, unique=True, index=True)

price = Column(Float)

orders = relationship("Order", back_populates="food")

class Order(Base):

__tablename__ = "orders"

id = Column(Integer, primary_key=True, index=True)

food_id = Column(Integer, ForeignKey("food.id"))

user_id = Column(Integer, ForeignKey("users.id"))

user = relationship("User", back_populates="orders")

food = relationship("Food", back_populates="orders")

DATABASE_URL = "sqlite:///example.db"

engine = create_engine(DATABASE_URL)

SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

def init_db():

Base.metadata.create_all(bind=engine)

session = SessionLocal()

users = [

User(name="Alice", age=30, email="alice@example.com"),

User(name="Bob", age=25, email="bob@example.com"),

User(name="Charlie", age=35, email="charlie@example.com"),

]

session.add_all(users)

session.commit()

foods = [

Food(name="Pizza Margherita", price=12.5),

Food(name="Spaghetti Carbonara", price=15.0),

Food(name="Lasagne", price=14.0),

]

session.add_all(foods)

session.commit()

orders = [

Order(food_id=1, user_id=1),

Order(food_id=2, user_id=1),

Order(food_id=3, user_id=2),

]

session.add_all(orders)

session.commit()

session.close()

print("Datenbank wurde erfolgreich erweitert und mit Beispieldaten gefüllt.")

if __name__ == "__main__":

if not os.path.exists("example.db"):

init_db()

else:

print("Die Datenbank 'example.db' existiert bereits.")