start with type annotation

This commit is contained in:
Patrick Kanzler 2020-11-08 23:45:25 +01:00
parent dd3c768f13
commit ed8ec0788a
1 changed files with 7 additions and 5 deletions

View File

@ -11,6 +11,8 @@ import yaml
from tempfile import gettempdir from tempfile import gettempdir
import platform import platform
from typing import Any, Dict
logging.basicConfig() logging.basicConfig()
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -45,7 +47,7 @@ if full_load:
logger.debug('Finished loading capabilities took %.2fs', time.time() - t0) logger.debug('Finished loading capabilities took %.2fs', time.time() - t0)
PROFILES = CAPABILITIES['profiles'] PROFILES: Dict[str, Any] = CAPABILITIES['profiles']
class NotSupported(Exception): class NotSupported(Exception):
@ -65,12 +67,12 @@ class BaseProfile(object):
features, colors and more. features, colors and more.
""" """
profile_data = {} profile_data: Dict[str, Any] = {}
def __getattr__(self, name): def __getattr__(self, name):
return self.profile_data[name] return self.profile_data[name]
def get_font(self, font): def get_font(self, font) -> int:
"""Return the escpos index for `font`. Makes sure that """Return the escpos index for `font`. Makes sure that
the requested `font` is valid. the requested `font` is valid.
""" """
@ -97,7 +99,7 @@ class BaseProfile(object):
return {v: k for k, v in self.codePages.items()} return {v: k for k, v in self.codePages.items()}
def get_profile(name=None, **kwargs): def get_profile(name:str=None, **kwargs):
"""Get the profile by name; if no name is given, return the """Get the profile by name; if no name is given, return the
default profile. default profile.
""" """
@ -111,7 +113,7 @@ def get_profile(name=None, **kwargs):
CLASS_CACHE = {} CLASS_CACHE = {}
def get_profile_class(name): def get_profile_class(name: str):
"""For the given profile name, load the data from the external """For the given profile name, load the data from the external
database, then generate dynamically a class. database, then generate dynamically a class.
""" """