glasswall.determine_file_type.helpers

 1from typing import Union
 2
 3from glasswall.determine_file_type.classes import FileTypeEnum, FileTypeEnumError, FileTypeEnumSuccess
 4
 5error_list = FileTypeEnumError.__subclasses__()
 6success_list = FileTypeEnumSuccess.__subclasses__()
 7
 8int_str_map = {
 9    fte_class.integer: fte_class.string
10    for fte_class in error_list + success_list
11    if all(key in fte_class.__dict__.keys() for key in ["integer", "string"])
12}
13
14str_int_map = {
15    fte_class.string: fte_class.integer
16    for fte_class in error_list + success_list
17    if all(key in fte_class.__dict__.keys() for key in ["integer", "string"])
18}
19
20int_class_map = {
21    fte_class.integer: fte_class
22    for fte_class in error_list + success_list
23    if all(key in fte_class.__dict__.keys() for key in ["integer"])
24}
25
26
27def is_success(file_type: Union[int, str, FileTypeEnumError, FileTypeEnumSuccess]):
28    """ Checks if a file type corresponds to a success.
29
30    Args:
31        file_type (Union[int, str, FileTypeEnumError, FileTypeEnumSuccess]): An enum int returned by Glasswall, str representation of file type, or subclass of FileTypeEnum.
32
33    Returns:
34        bool: Returns True if file_type corresponds to a successful file type, else False.
35    """
36    if isinstance(file_type, int):
37        return file_type in [s.integer for s in success_list]
38    elif isinstance(file_type, str):
39        return file_type in [s.string for s in success_list]
40    elif issubclass(file_type, FileTypeEnum):
41        return issubclass(file_type, FileTypeEnumSuccess)
42    else:
43        raise TypeError(file_type)
44
45
46def file_type_int_to_str(integer: int):
47    """ Converts a file type enum int to a string.
48
49    Args:
50        integer (int): The enum int that Glasswall returns when ing a file type.
51
52    Returns:
53        Union[type(None), str]: The string representation of a file type, or None. """
54    return int_str_map.get(integer, None)
55
56
57def file_type_str_to_int(string: str):
58    """ Converts a file type string to an enum int.
59
60    Args:
61        string (str): A string representation of a file type.
62
63    Returns:
64        Union[type(None), int]: The enum int that Glasswall returns when determining a file type, or None. """
65    return str_int_map.get(string, None)