fix: improve the type system

Author: tmgbedu

fastapi_startkit/.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.languageServer": "None"
+}

fastapi_startkit/src/fastapi_startkit/masoniteorm/models/model.py

@@ -280,7 +280,7 @@ def new_model_instance(self, attributes=None, exists=False):
 
         return model
 
-    def new_query(self):
+    def new_query(self) -> "QueryBuilder":
         return self.db_manager.connection(self.connection).query().set_model(self)
 
     def hydrate(self, items):
@@ -310,7 +310,7 @@ def __getattr__(self, attribute):
         return self.get_attribute(attribute)
 
     @classmethod
-    def query(cls):
+    def query(cls) -> "QueryBuilder":
         return cls().new_query()
 
     @classmethod
@@ -402,13 +402,5 @@ def get_attributes(self) -> dict:
     def serialize(self) -> dict:
         return self.get_attributes()
 
-    @classmethod
-    def where(cls, column, *args):
-        return cls().query().where(column, *args)
-
-    @classmethod
-    def where_in(cls, column, values):
-        return cls().query().where_in(column, values)
-
     def get_table_name(self):
         return self.__table__ or inflection.tableize(self.__class__.__name__)