add type annotation for id (#197)

* add type annotation for `id` * do the same for `TokenAuth` table
piccolo-orm · Oct 18, 2022 · aa6fadb · aa6fadb
1 parent 99077f5
commit aa6fadb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/piccolo_api/session_auth/tables.py b/piccolo_api/session_auth/tables.py
@@ -4,7 +4,7 @@
 import typing as t
 from datetime import datetime, timedelta
 
-from piccolo.columns import Integer, Timestamp, Varchar
+from piccolo.columns import Integer, Serial, Timestamp, Varchar
 from piccolo.columns.defaults.timestamp import TimestampOffset
 from piccolo.table import Table
 from piccolo.utils.sync import run_sync
@@ -15,6 +15,8 @@ class SessionsBase(Table, tablename="sessions"):
     Use this table, or inherit from it, to create a session store.
     """
 
+    id: Serial
+
     #: Stores the session token.
     token: Varchar = Varchar(length=100, null=False)
 

diff --git a/piccolo_api/token_auth/tables.py b/piccolo_api/token_auth/tables.py
@@ -4,7 +4,7 @@
 import uuid
 
 from piccolo.apps.user.tables import BaseUser
-from piccolo.columns.column_types import ForeignKey, Varchar
+from piccolo.columns.column_types import ForeignKey, Serial, Varchar
 from piccolo.table import Table
 from piccolo.utils.sync import run_sync
 
@@ -24,6 +24,7 @@ class TokenAuth(Table):
     web usage.
     """
 
+    id: Serial
     token = Varchar(default=generate_token)
     user = ForeignKey(references=BaseUser)