add more type notations

nitbix · Sep 13, 2020 · 58f2c64 · 58f2c64
1 parent 24b56a4
commit 58f2c64
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/toupee/config.py b/toupee/config.py
@@ -48,7 +48,7 @@ def include(self, node):
 
 Loader.add_constructor('!include', Loader.include)
 
-def load_parameters(filename):
+def load_parameters(filename: str) -> parameters.Parameters:
     """ Loads a parameters file """
     with open(filename) as f:
         r = yaml.load(f)

diff --git a/toupee/parameters.py b/toupee/parameters.py
@@ -10,7 +10,7 @@
 
 import copy
 
-to_string = ['method']
+__STRINGABLE = ['method']
 class Parameters(object):
 
     def __init__(self, **entries):
@@ -19,7 +19,7 @@ def __init__(self, **entries):
     def serialize(self):
         serialized = copy.deepcopy(self)
 
-        for x in to_string:
+        for x in __STRINGABLE:
             if x in serialized.__dict__:
                 serialized.__dict__[x] = serialized.__dict__[x].serialize()
         return serialized.__dict__