Add clone, referring, and traffic sqlmodels

Author: astrochun

github_stats_pages/models/__init__.py

@@ -0,0 +1,3 @@
+from .clone import Clone  # noqa: F401
+from .referring import Referring  # noqa: F401
+from .traffic import Traffic  # noqa: F401

github_stats_pages/models/clone.py

@@ -0,0 +1,11 @@
+from typing import Optional
+
+from sqlmodel import SQLModel, Field
+
+
+class Clone(SQLModel, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True)
+    repository_name: str
+    date: str
+    clones: int
+    unique_clones: int

github_stats_pages/models/referring.py

@@ -0,0 +1,12 @@
+from typing import Optional
+
+from sqlmodel import SQLModel, Field
+
+
+class Referring(SQLModel, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True)
+    repository_name: str
+    site: str
+    date: str
+    views: int
+    unique_visitors: int

github_stats_pages/models/traffic.py

@@ -0,0 +1,11 @@
+from typing import Optional
+
+from sqlmodel import SQLModel, Field
+
+
+class Traffic(SQLModel, table=True):
+    id: Optional[int] = Field(default=None, primary_key=True)
+    repository_name: str
+    date: str
+    views: int
+    unique_visitors: int

setup.cfg

@@ -37,6 +37,7 @@ install_requires =
     PyGithub == 1.55
     tabulate == 0.8.7
     rich >= 12.4.1, <13.0.0
+    sqlmodel >= 0.0.6, < 1.0.0
 scripts =
     scripts/get_repo_list
     scripts/gts_run_all_repos