Browse Source

Fix some types

Getty Ritter 4 years ago
parent
commit
788d7e5648
3 changed files with 6 additions and 11 deletions
  1. 2 2
      lc/model.py
  2. 1 9
      lc/web.py
  3. 3 0
      stubs/playhouse/shortcuts.py

+ 2 - 2
lc/model.py

@@ -13,7 +13,7 @@ class Model(peewee.Model):
     class Meta:
         database = c.DB
 
-    def to_dict(self):
+    def to_dict(self) -> dict:
         return playhouse.shortcuts.model_to_dict(self)
 
 
@@ -60,7 +60,7 @@ class User(Model):
     def get_tag(self, tag_name: str) -> "Tag":
         return Tag.get((Tag.user == self) & (Tag.name == tag_name))
 
-    def to_dict(self):
+    def to_dict(self) -> dict:
         return {"id": self.id, "name": self.name}
 
 

+ 1 - 9
lc/web.py

@@ -2,6 +2,7 @@ import flask
 import pystache
 
 import lc.config as c
+import lc.error as e
 import lc.model as m
 
 
@@ -41,15 +42,6 @@ class Endpoint:
         if not self.user or name != self.user.name:
             raise e.BadPermissions()
 
-    def api_post(self, *args, **kwargs) -> dict:
-        raise e.NotImplemented()
-
-    def api_get(self, *args, **kwargs) -> dict:
-        raise e.NotImplemented()
-
-    def html(self, *args, **kwargs):
-        raise e.NotImplemented()
-
     def route(self, *args, **kwargs):
         try:
             if flask.request.method == "POST":

+ 3 - 0
stubs/playhouse/shortcuts.py

@@ -0,0 +1,3 @@
+import peewee
+
+def model_to_dict(m: peewee.Model) -> dict: pass