|
1 | 1 | """Declare models and relationships.""" |
2 | | -from sqlalchemy import ( |
3 | | - Boolean, |
4 | | - Column, |
5 | | - DateTime, |
6 | | - ForeignKey, |
7 | | - Integer, |
8 | | - String, |
9 | | - Table, |
10 | | - Text, |
11 | | - text, |
12 | | -) |
| 2 | +from sqlalchemy import Boolean, Column, DateTime, ForeignKey, Integer, String, Text |
13 | 3 | from sqlalchemy.ext.declarative import declarative_base |
14 | 4 | from sqlalchemy.orm import relationship |
| 5 | +from sqlalchemy.sql import func |
| 6 | + |
15 | 7 | from sqlalchemy_tutorial.database import engine |
16 | 8 |
|
17 | 9 | Base = declarative_base() |
18 | 10 |
|
19 | | -association_table = Table( |
20 | | - "association", |
21 | | - Base.metadata, |
22 | | - Column("team_id", Integer, ForeignKey("sport.player.team_id")), |
23 | | - Column("id", Integer, ForeignKey("sport.team.id")), |
24 | | -) |
25 | | - |
26 | 11 |
|
27 | 12 | class Player(Base): |
28 | | - """Individual player object.""" |
| 13 | + """Individual player belonging to a team.""" |
29 | 14 |
|
30 | 15 | __tablename__ = "player" |
31 | 16 |
|
32 | 17 | id = Column(Integer, primary_key=True, autoincrement="auto") |
33 | | - team_id = Column(Integer, ForeignKey("sport.team.id"), nullable=False) |
| 18 | + team_id = Column(Integer, ForeignKey("team.id"), nullable=False) |
34 | 19 | first_name = Column(String(255), nullable=False) |
35 | 20 | last_name = Column(String(255), nullable=False) |
36 | 21 | position = Column(String(100), nullable=False) |
37 | 22 | injured = Column(Boolean) |
38 | 23 | description = Column(Text, nullable=True) |
39 | | - created_at = Column(DateTime) |
40 | | - updated_at = Column(DateTime, server_default=text("NOW()"), nullable=False) |
| 24 | + created_at = Column(DateTime, server_default=func.now()) |
| 25 | + updated_at = Column(DateTime, onupdate=func.now()) |
41 | 26 |
|
42 | 27 | # Relationships |
43 | 28 | team = relationship("Team") |
44 | 29 |
|
45 | 30 | def __repr__(self): |
46 | | - return "<Person model {}>".format(self.id) |
| 31 | + return "<Player {}>".format(self.id) |
47 | 32 |
|
48 | 33 |
|
49 | 34 | class Team(Base): |
50 | | - """Sport team consisting of players.""" |
| 35 | + """Team consisting of many players.""" |
51 | 36 |
|
52 | 37 | __tablename__ = "team" |
53 | 38 |
|
54 | 39 | id = Column(Integer, primary_key=True, autoincrement="auto") |
55 | 40 | name = Column(String(255), nullable=False) |
56 | 41 | city = Column(String(255), nullable=False) |
57 | | - created_at = Column(DateTime, nullable=False) |
58 | | - updated_at = Column(DateTime, server_default=text("NOW()"), nullable=False) |
| 42 | + abbreviation = Column(String(255), nullable=False) |
| 43 | + logo = Column(String(255), nullable=False) |
| 44 | + created_at = Column(DateTime, server_default=func.now()) |
| 45 | + updated_at = Column(DateTime, onupdate=func.now()) |
59 | 46 |
|
60 | 47 | def __repr__(self): |
61 | | - return "<Team model {}>".format(self.id) |
| 48 | + return "<Team {}>".format(self.id) |
62 | 49 |
|
63 | 50 |
|
64 | 51 | def create_tables(): |
65 | 52 | return Base.metadata.create_all(engine) |
66 | | - |
0 commit comments