diff options
Diffstat (limited to 'dbstore')
-rw-r--r-- | dbstore/peewee_store.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/dbstore/peewee_store.py b/dbstore/peewee_store.py index 37b8c01..00897d7 100644 --- a/dbstore/peewee_store.py +++ b/dbstore/peewee_store.py | |||
@@ -1,5 +1,10 @@ | |||
1 | from peewee import * | 1 | from peewee import * |
2 | 2 | ||
3 | TOOT_VISIBILITY_PUBLIC = "public" | ||
4 | TOOT_VISIBILITY_UNLISTED = "unlisted" | ||
5 | TOOT_VISIBILITY_PRIVATE = "private" | ||
6 | |||
7 | |||
3 | db = SqliteDatabase("checkinbot.db") | 8 | db = SqliteDatabase("checkinbot.db") |
4 | db.connect(reuse_if_open=True) | 9 | db.connect(reuse_if_open=True) |
5 | 10 | ||
@@ -16,6 +21,7 @@ class User(BaseModel): | |||
16 | state = CharField(max_length=128) | 21 | state = CharField(max_length=128) |
17 | client_id = CharField(max_length=128) | 22 | client_id = CharField(max_length=128) |
18 | client_secret = CharField(max_length=128) | 23 | client_secret = CharField(max_length=128) |
24 | toot_visibility = CharField(max_length=128, default=TOOT_VISIBILITY_PRIVATE) | ||
19 | 25 | ||
20 | 26 | ||
21 | class Location(BaseModel): | 27 | class Location(BaseModel): |