peewee_orm_tutorial/models.py at main · Thebored1/peewee_orm_tutorial · GitHub

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
import peewee
import datetime

database = peewee.SqliteDatabase("test.db")

class Students(peewee.Model):
    """
    ORM model of the Students table
    """
    name = peewee.CharField()
    admission = peewee.DateField(default=datetime.date.today)


    class Meta:
        database = database


class Schools(peewee.Model):
    """
    ORM model of Schools table
    """
    school = peewee.CharField()
    location = peewee.CharField()

    class Meta:
        database = database


if __name__ == "__main__":
    """
    Handling exceptions
    """
    try:
        Students.create_table()
    except peewee.OperationalError:
        print ("Students table already exists!")

    try:
        Schools.create_table()
    except peewee.OperationalError:
        print ("Schools table already exists!")