cleanups, fix settings

mypy_django_plugin/plugins/related_fields.py

@@ -1,18 +1,12 @@
 import typing
 from typing import Optional, cast
 
-from django.conf import Settings
 from mypy.checker import TypeChecker
-from mypy.nodes import MDEF, AssignmentStmt, MypyFile, StrExpr, TypeInfo, NameExpr, Var, SymbolTableNode
-from mypy.plugin import FunctionContext, ClassDefContext
+from mypy.nodes import StrExpr
+from mypy.plugin import FunctionContext
 from mypy.types import Type, CallableType, Instance, AnyType, TypeOfAny
 
 from mypy_django_plugin import helpers
-from mypy_django_plugin.helpers import get_models_file
-
-
-def extract_related_name_value(ctx: FunctionContext) -> str:
-    return ctx.args[ctx.arg_names.index('related_name')][0].value
 
 
 def reparametrize_with(instance: Instance, new_typevars: typing.List[Type]):
@@ -55,44 +49,9 @@ def get_valid_to_value_or_none(ctx: FunctionContext) -> Optional[Instance]:
     return referred_to_type
 
 
-def add_new_var_node_to_class(class_type: TypeInfo, name: str, typ: Instance) -> None:
-    var = Var(name=name, type=typ)
-    var.info = typ.type
-    var._fullname = class_type.fullname() + '.' + name
-    var.is_inferred = True
-    var.is_initialized_in_class = True
-    class_type.names[name] = SymbolTableNode(MDEF, var)
-
-
-def extract_to_parameter_as_get_ret_type(ctx: FunctionContext) -> Type:
+def extract_to_parameter_as_get_ret_type_for_related_field(ctx: FunctionContext) -> Type:
     referred_to_type = get_valid_to_value_or_none(ctx)
     if referred_to_type is None:
         # couldn't extract to= value
         return fill_typevars_with_any(ctx.default_return_type)
     return reparametrize_with(ctx.default_return_type, [referred_to_type])
-
-
-def set_fieldname_attrs_for_related_fields(ctx: ClassDefContext) -> None:
-    api = ctx.api
-    for stmt in ctx.cls.defs.body:
-        if not isinstance(stmt, AssignmentStmt):
-            continue
-        if not hasattr(stmt.rvalue, 'callee'):
-            continue
-        if len(stmt.lvalues) > 1:
-            # multiple lvalues not supported for now
-            continue
-
-        expr = stmt.lvalues[0]
-        if not isinstance(expr, NameExpr):
-            continue
-        name = expr.name
-
-        rvalue_callee = stmt.rvalue.callee
-        if rvalue_callee.fullname in {helpers.FOREIGN_KEY_FULLNAME,
-                                      helpers.ONETOONE_FIELD_FULLNAME}:
-            name += '_id'
-            new_node = helpers.create_new_symtable_node(name,
-                                                        kind=MDEF,
-                                                        instance=api.named_type('__builtins__.int'))
-            ctx.cls.info.names[name] = new_node