mirror of
https://github.com/mxrch/GHunt.git
synced 2026-10-24 01:29:24 +08:00
mxrch
39 lines
1.8 KiB
Python
39 lines
1.8 KiB
Python
from ghunt.knowledge.services import services_baseurls
|
|
from ghunt.knowledge.keys import keys
|
|
from ghunt.knowledge.maps import types_translations
|
|
from ghunt.knowledge.people import user_types
|
|
from ghunt.knowledge.sig import sigs
|
|
from ghunt.errors import GHuntKnowledgeError
|
|
|
|
from typing import *
|
|
|
|
|
|
def get_domain_of_service(service: str) -> str:
|
|
if service not in services_baseurls:
|
|
raise GHuntKnowledgeError(f'The service "{service}" has not been found in GHunt\'s services knowledge.')
|
|
return services_baseurls.get(service)
|
|
|
|
def get_origin_of_key(key_name: str) -> str:
|
|
if key_name not in keys:
|
|
raise GHuntKnowledgeError(f'The key "{key_name}" has not been found in GHunt\'s API keys knowledge.')
|
|
return keys.get(key_name, {}).get("origin")
|
|
|
|
def get_api_key(key_name: str) -> str:
|
|
if key_name not in keys:
|
|
raise GHuntKnowledgeError(f'The key "{key_name}" has not been found in GHunt\'s API keys knowledge.')
|
|
return keys.get(key_name, {}).get("key")
|
|
|
|
def get_gmaps_type_translation(type_name: str) -> str:
|
|
if type_name not in types_translations:
|
|
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 !')
|
|
return types_translations.get(type_name)
|
|
|
|
def get_user_type_definition(type_name: str) -> str:
|
|
if type_name not in user_types:
|
|
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 !')
|
|
return user_types.get(type_name)
|
|
|
|
def get_package_sig(package_name: str) -> str:
|
|
if package_name not in sigs:
|
|
raise GHuntKnowledgeError(f'The package name "{package_name}" has not been found in GHunt\'s SIGs knowledge.')
|
|
return sigs.get(package_name) |