1 # Fedi API Block - An aggregator for fetching blocking data from fediverse nodes
2 # Copyright (C) 2023 Free Software Foundation
4 # This program is free software: you can redistribute it and/or modify
5 # it under the terms of the GNU Affero General Public License as published
6 # by the Free Software Foundation, either version 3 of the License, or
7 # (at your option) any later version.
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU Affero General Public License for more details.
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <https://www.gnu.org/licenses/>.
20 logging.basicConfig(level=logging.INFO)
21 logger = logging.getLogger(__name__)
23 # Cache for redundant SQL queries
28 def key_exists(key: str) -> bool:
29 logger.debug("key='%s' - CALLED!", key)
30 exists = key in _cache
32 logger.debug("exists='%s' - EXIT!", exists)
35 def set_all(key: str, rows: list, value: any):
36 logger.debug("key='%s',rows()=%d,value[]='%s' - CALLED!", key, len(rows), type(value))
38 if not isinstance(key, str):
39 raise ValueError(f"Parameter key[]='{type(key)}' is not of type 'str'")
40 elif not key_exists(key):
41 logger.debug("Cache for key='%s' not initialized.", key)
44 logger.debug("Setting %d row(s) for key='%s',value[%s]='%s' ...", len(rows), key, type(value), value)
46 logger.debug("Setting key='%s',sub[%s]='%s'", key, type(sub), sub)
47 if isinstance(sub, sqlite3.Row):
48 logger.debug("Setting key='%s',sub[%s]='%s',value[]='%s'", key, type(sub), sub, type(value))
49 _cache[key][sub[0]] = value
51 logger.warning("Unsupported type sub[]='%s'", type(sub))
55 def set_sub_key(key: str, sub: str, value: any):
56 logger.debug("key='%s',sub='%s',value[]='%s' - CALLED!", key, sub, type(value))
58 if not isinstance(key, str):
59 raise ValueError(f"Parameter key[]='{type(key)}' is not of type 'str'")
60 elif not isinstance(sub, str):
61 raise ValueError(f"Parameter sub[]='{type(sub)}' is not of type 'str'")
62 elif not key_exists(key):
63 raise Exception(f"Cache for key='{key}' is not initialized, but function invoked")
65 logger.debug("Setting key='%s',sub='%s',value[%s]='%s' ...", key, sub, type(value), value)
66 _cache[key][sub] = value
70 def sub_key_exists(key: str, sub: str) -> bool:
71 logger.debug("key='%s',sub='%s' - CALLED!", key, sub)
73 if not isinstance(key, str):
74 raise ValueError(f"Parameter key[]='{type(key)}' is not of type 'str'")
75 elif not isinstance(sub, str):
76 raise ValueError(f"Parameter sub[]='{type(sub)}' is not of type 'str'")
77 elif not key_exists(key):
78 raise Exception(f"Cache for key='{key}' is not initialized, but function invoked")
80 exists = sub in _cache[key]
82 logger.debug("exists='%s' - EXIT!", exists)