meca-restapi-managers-token 1.0.0__tar.gz

meca_restapi_managers_token-1.0.0/MANIFEST.in

@@ -0,0 +1,6 @@
+# https://setuptools.pypa.io/en/latest/userguide/miscellaneous.html#using-manifest-in
+prune .vscode
+prune docs
+prune Pipfile
+prune Pipfile.lock
+prune tests

meca_restapi_managers_token-1.0.0/PKG-INFO

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: meca-restapi-managers-token
+Version: 1.0.0
+Summary: 令牌管理接口
+Author: dqlr
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.135.1
+Requires-Dist: sqlmodel>=0.0.37
+Requires-Dist: uvicorn>=0.41.0
+
+# 

meca_restapi_managers_token-1.0.0/README.md

@@ -0,0 +1 @@
+# 

meca_restapi_managers_token-1.0.0/meca_restapi_managers_token.egg-info/PKG-INFO

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: meca-restapi-managers-token
+Version: 1.0.0
+Summary: 令牌管理接口
+Author: dqlr
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: fastapi>=0.135.1
+Requires-Dist: sqlmodel>=0.0.37
+Requires-Dist: uvicorn>=0.41.0
+
+# 

meca_restapi_managers_token-1.0.0/meca_restapi_managers_token.egg-info/SOURCES.txt

@@ -0,0 +1,14 @@
+MANIFEST.in
+README.md
+pyproject.toml
+meca_restapi_managers_token.egg-info/PKG-INFO
+meca_restapi_managers_token.egg-info/SOURCES.txt
+meca_restapi_managers_token.egg-info/dependency_links.txt
+meca_restapi_managers_token.egg-info/requires.txt
+meca_restapi_managers_token.egg-info/top_level.txt
+src/__init__.py
+src/routers/__init__.py
+src/routers/_token.py
+src/tables/__init__.py
+src/tables/db.py
+src/tables/token.py

meca_restapi_managers_token-1.0.0/meca_restapi_managers_token.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

meca_restapi_managers_token-1.0.0/meca_restapi_managers_token.egg-info/requires.txt

@@ -0,0 +1,3 @@
+fastapi>=0.135.1
+sqlmodel>=0.0.37
+uvicorn>=0.41.0

meca_restapi_managers_token-1.0.0/meca_restapi_managers_token.egg-info/top_level.txt

@@ -0,0 +1 @@
+meca

meca_restapi_managers_token-1.0.0/pyproject.toml

@@ -0,0 +1,21 @@
+[build-system]
+requires        = ['setuptools>=70.0']
+build-backend   = "setuptools.build_meta"
+
+[project]
+name    = "meca-restapi-managers-token"
+version = "1.0.0"
+authors = [
+    { name = "dqlr" }
+]
+description = "令牌管理接口"
+requires-python = ">=3.10"
+readme = 'README.md'
+dependencies = [
+    "fastapi >=0.135.1",
+    "sqlmodel >=0.0.37",
+    "uvicorn >=0.41.0"
+]
+
+[tool.setuptools.package-dir]
+"meca.restapi.managers.token" = "src"

meca_restapi_managers_token-1.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

meca_restapi_managers_token-1.0.0/src/__init__.py

@@ -0,0 +1,5 @@
+from .routers import token as token_router
+
+__all__ = [
+    'token_router'
+]

meca_restapi_managers_token-1.0.0/src/routers/__init__.py

@@ -0,0 +1,5 @@
+from ._token import token
+
+__all__ = [
+    "token"
+]

meca_restapi_managers_token-1.0.0/src/routers/_token.py

@@ -0,0 +1,29 @@
+from ..tables.db import SqlDB
+from ..tables.token import Token
+
+from fastapi import APIRouter
+
+db = SqlDB('home#token.db').wal().check_same_thread(False)
+SessionAnnotated = db.SessionAnnotated
+token = APIRouter(lifespan=db.init)
+
+
+@token.post('/update')
+async def update(token: Token, session: SessionAnnotated) -> Token:
+    existing = session.get(Token, (token.app, token.user))
+    if existing:
+        session.merge(token)
+        session.commit()
+        session.refresh(existing)
+        return existing
+    else:
+        session.add(token)
+        session.commit()
+        session.refresh(token)
+        return token
+
+
+@token.post('/get')
+def get(token: Token, session: SessionAnnotated) -> Token:
+    token = session.get(Token, (token.app, token.user))
+    return token

meca_restapi_managers_token-1.0.0/src/tables/db.py

@@ -0,0 +1,70 @@
+from contextlib import asynccontextmanager
+from pathlib import Path
+from typing import Annotated, List
+from typing_extensions import Self  # 兼容Python3.10
+
+from sqlalchemy import Engine
+from sqlmodel import Session, SQLModel, create_engine
+
+from fastapi import Depends, FastAPI
+
+
+class SqlDB:
+    path: Path | str
+    connect_args: dict = {}
+    uri: str
+    query: List[str] = []
+    db: str
+
+    _engine: Engine = None
+
+    def _parse_path(self, path: str | Path):
+        if isinstance(path, Path):
+            path = path.absolute()
+        elif isinstance(path, str) and path.startswith('home#'):
+            path = (Path('~')/'databases'/path.split('#')[1]).expanduser().absolute()
+        else:
+            return
+
+        if not path.parent.is_dir():
+            path.parent.mkdir(parents=True, exist_ok=True)
+        self.uri = f'sqlite:///{path.as_posix()}'
+        self.path = path
+
+    def __init__(self, path: str | Path) -> Self:
+        self._parse_path(path=path)
+
+    def __str__(self):
+        if self.query:
+            return f'{self.uri}?{"&".join(self.query)}'
+        else:
+            return f'{self.uri}'
+        
+    def __repr__(self):
+        return str(self)
+    
+    def wal(self) -> Self:
+        self.query.append('journal_mode=WAL')
+        return self
+    
+    def check_same_thread(self, value: bool) -> Self:
+        self.connect_args['check_same_thread'] = value
+        return self
+
+    @property
+    def engine(self) -> Engine:
+        if self._engine is None:
+            self._engine = create_engine(str(self), connect_args=self.connect_args)
+        return self._engine
+
+    @asynccontextmanager
+    async def init(self, _app: FastAPI):
+        SQLModel.metadata.create_all(self.engine)
+        yield
+
+    @property
+    def SessionAnnotated(self) -> Annotated:
+        def _session():
+            with Session(self.engine) as __session:
+                yield __session
+        return Annotated[Session, Depends(_session)]

meca_restapi_managers_token-1.0.0/src/tables/token.py

@@ -0,0 +1,18 @@
+from sqlmodel import Field, SQLModel
+from time import time
+
+class Token(SQLModel, table=True):
+    app: str = Field(primary_key=True)
+    user: str = Field(primary_key=True)
+    access_token: str
+    refresh_token: str
+    access_exp: int
+    refresh_exp: int
+
+    @property
+    def access_valid(self) -> bool:
+        return self.access_exp > time.time()
+    
+    @property
+    def refresh_valid(self) -> bool:
+        return self.refresh_exp > time.time()