initial commit

app.py

@@ -0,0 +1,84 @@
+import typing
+from typing import List, Dict
+
+from fastapi import FastAPI, Depends
+from pydantic import BaseModel
+from starlette.middleware.cors import CORSMiddleware
+from starlette.requests import Request
+from starlette.responses import Response, JSONResponse
+import httpx
+
+from anki import AnkiClient, CookieStorage
+
+app = FastAPI(
+    title='anki',
+    docs_url='/')
+app.add_middleware(CORSMiddleware, allow_origins=['*'], allow_methods=['*'], allow_headers=['*'])
+
+
+class SessionCookieStorage(CookieStorage):
+    def __init__(self, req: Request, res: Response):
+        self.res = res
+        self.req = req
+
+    def save_cookies(self, cookies: dict):
+        for k, v in cookies.items():
+            self.res.set_cookie(k, v, max_age=24 * 60 * 60)
+
+    def load_cookies(self) -> typing.Optional[dict]:
+        if 'ankiweb' not in self.req.cookies:
+            return None
+        return {**self.req.cookies}
+
+
+def get_anki_client(req: Request, res: Response) -> AnkiClient:
+    return AnkiClient(SessionCookieStorage(req, res))
+
+
+@app.post('/login', response_model=Dict[str, str])
+async def login(username: str, password: str, anki: AnkiClient = Depends(get_anki_client)):
+    anki.login(username, password)
+    return anki.session
+
+
+@app.get('/info', summary='Get note types and their fields', response_model=List[Dict])
+async def info(anki: AnkiClient = Depends(get_anki_client)):
+    return anki.get_editor_context()
+
+
+class CreateNote(BaseModel):
+    deck: str
+    note_type: str
+    note_fields: Dict[str, str]
+    tags: str = ''
+
+
+@app.post('/notes', summary='Create a note')
+async def create_note(input: CreateNote, anki: AnkiClient = Depends(get_anki_client)):
+    anki.create_note(note_type=input.note_type,
+                     deck=input.deck,
+                     fields=input.note_fields,
+                     tags=input.tags)
+
+
+@app.exception_handler(Exception)
+async def handle_errors(req: Request, error: Exception):
+    if isinstance(error, httpx.HTTPStatusError):
+        message = 'anki returned error'
+        status = error.response.status_code
+    elif isinstance(error, PermissionError):
+        message = str(error)
+        status = 401
+    else:
+        message = 'oops'
+        status = 500
+
+    return JSONResponse(content={
+        'message': message
+    }, status_code=status)
+
+
+if __name__ == '__main__':
+    import uvicorn
+
+    uvicorn.run(app, host='0.0.0.0')