Skip to content
Snippets Groups Projects
provider.py 1.04 KiB
Newer Older
  • Learn to ignore specific revisions
  • Kali Kaneko's avatar
    Kali Kaneko committed
    import datetime
    import os
    
    
    def getDefaultProvider(config):
    
        if os.environ.get('VENDOR_PATH'):
    
    Kali Kaneko's avatar
    Kali Kaneko committed
            print('[+] Using default provider from config file')
            provider = config['default']['provider']
    
        else:
            provider = os.environ.get('PROVIDER')
            if provider:
                print('[+] Got provider {} from environment'.format(provider))
    
    Kali Kaneko's avatar
    Kali Kaneko committed
        return provider
    
    
    def getProviderData(provider, config):
        print("[+] Configured provider:", provider)
    
        try:
            c = config[provider]
        except Exception:
            raise ValueError('Cannot find provider')
            
    
    Kali Kaneko's avatar
    Kali Kaneko committed
        d = dict()
    
        keys = ('name', 'applicationName', 'binaryName', 'auth', 'authEmptyPass',
    
    Kali Kaneko's avatar
    Kali Kaneko committed
                'providerURL', 'tosURL', 'helpURL',
    
                'askForDonations', 'donateURL', 'apiURL',
                'geolocationAPI', 'caCertString')
    
    Kali Kaneko's avatar
    Kali Kaneko committed
    
        for value in keys:
            d[value] = c.get(value)
    
            if value == 'askForDonations':
                d[value] = bool(d[value])
    
    Kali Kaneko's avatar
    Kali Kaneko committed
    
        d['timeStamp'] = '{:%Y-%m-%d %H:%M:%S}'.format(
            datetime.datetime.now())
    
        return d