1
from ghunt.knowledge.services import services_baseurls
2
from ghunt.knowledge.keys import keys
3
from ghunt.knowledge.maps import types_translations
4
from ghunt.knowledge.people import user_types
5
from ghunt.knowledge.sig import sigs
6
from ghunt.errors import GHuntKnowledgeError
7

8
from typing import *
9

10

11
def get_domain_of_service(service: str) -> str:
12
    if service not in services_baseurls:
13
        raise GHuntKnowledgeError(f'The service "{service}" has not been found in GHunt\'s services knowledge.')
14
    return services_baseurls.get(service)
15

16
def get_origin_of_key(key_name: str) -> str:
17
    if key_name not in keys:
18
        raise GHuntKnowledgeError(f'The key "{key_name}" has not been found in GHunt\'s API keys knowledge.')
19
    return keys.get(key_name, {}).get("origin")
20

21
def get_api_key(key_name: str) -> str:
22
    if key_name not in keys:
23
        raise GHuntKnowledgeError(f'The key "{key_name}" has not been found in GHunt\'s API keys knowledge.')
24
    return keys.get(key_name, {}).get("key")
25

26
def get_gmaps_type_translation(type_name: str) -> str:
27
    if type_name not in types_translations:
28
        raise GHuntKnowledgeError(f'The gmaps type "{type_name}" has not been found in GHunt\'s knowledge.\nPlease open an issue on the GHunt Github or submit a PR to add it !')
29
    return types_translations.get(type_name)
30

31
def get_user_type_definition(type_name: str) -> str:
32
    if type_name not in user_types:
33
        raise GHuntKnowledgeError(f'The user type "{type_name}" has not been found in GHunt\'s knowledge.\nPlease open an issue on the GHunt Github or submit a PR to add it !')
34
    return user_types.get(type_name)
35

36
def get_package_sig(package_name: str) -> str:
37
    if package_name not in sigs:
38
        raise GHuntKnowledgeError(f'The package name "{package_name}" has not been found in GHunt\'s SIGs knowledge.')
39
    return sigs.get(package_name)
40