CHEM = "etizolam"   # Chemical of interest
REGION = "ALL"      # Country or "UK", use "ALL" for everything
    
# Possible regions: "Canada", "China", "CzechRepublic", "Germany", 
#                   "Hungary", "Japan", "Netherlands", "Poland", 
#                   "Spain", "Sweden", "UK", "USA", "ALL"

########################################
# IGNORE CODE BELOW
########################################


urls = {'Canada': ['biochemlabsupply.com',
            'chemscanada.com',
            'lysergi.com',
            'rechem.ca',
            'syntharise.com'],
 'China': ['benzochemicals.com',
           'cillinchem.com',
           'cwresc.org',
           'dharmachems.com',
           'kk-pu.com',
           'lsresearchchemlab.com',
           'reachchem.org',
           'rechemco.net',
           'tpr-power.com',
           'tripam.in'],
 'CzechRepublic': ['synthetiq.org',
                   'btgwholesale.ru',
                   'byindustries.info',
                   'etizolamb.com',
                   'plantfoodusa.net',
                   'purpleghostchemicals.com',
                   'qualityetizolamusa.com',
                   'downlowlabs.com',
                   'dllsupply.com',
                   'themonkeybiz.com'],
 'Germany': ['fox-chem.com'],
 'Hungary': ['rchems.org'],
 'Japan': ['mimaki-family-japan.com'],
 'Netherlands': ['buckledbonzi.com', 'new-forms.org'],
 'Poland': ['3mmc.org',
            'astro-lab.com',
            'bazarr.nl',
            'blackchem.com',
            'bobbyshop.eu',
            'buy-mmc.com',
            'dopking.pl',
            'fakeweed.nl',
            'genuine-rc.nl',
            'kolekcjoner.nl',
            'legalchem.pl',
            'lizardlabs.eu',
            'mefedron.pl',
            'rc-chemicals.cz',
            'rc-chemicals.nl',
            'safelab.es',
            'sun-fish.biz'],
 'Spain': ['buy-jwh.com', 'scientificinnovation.com'],
 'Sweden': ['aromer.nu',
            'asklepios.nu',
            'chemicalservices.org',
            'chemlegal.net',
            'hydrachems.se',
            'lateralus.se',
            'n2p0.com',
            'nordicrc.com',
            'oppiosite.name',
            'rc24.se',
            'rcbolaget.com',
            'swedenchems.com',
            'thetripp.net',
            'thesmack.biz'],
 'UK': ['blot-research.co.uk',
        'brc-finechemicals.com',
        'buckledbonzi.co.uk',
        'buylegalrc.eu',
        'cheapresearchchems.com',
        'chem-shop.co.uk',
        'buythegram.co.uk',
        'chemicalpowdershop.co.uk',
        'chemicalwire.com',
        'chemsupply.eu',
        'chemtheory.com',
        'cthighstreet.com',
        'gr8researchchemicals.com',
        'labtrade.co.uk',
        'lizardlabs.co.uk',
        'officialbenzofury.net',
        'organic-synth.com',
        'polatzoheadshop.co.uk',
        'purechemicals.net',
        'ravegardener.org',
        'researchchemicals4u.com',
        'researchchemist.co.uk',
        'researchchemistry.co.uk',
        'sciencesuppliesdirect.com',
        'seriousmolecule.com',
        'smokeyschemsite.com',
        'thercshop.org',
        'tkkchems.com',
        'ukrc-chemicals.co.uk'],
 'USA': ['btgwholesale.ru',
         'byindustries.info',
         'etizolamb.com',
         'plantfoodusa.net',
         'purpleghostchemicals.com',
         'qualityetizolamusa.com',
         'domestic-sync.net',
         'downlowlabs.com',
         'themonkeybiz.com']}

def flatten(x):
    result = []
    for el in x:
        if hasattr(el, "__iter__") and not isinstance(el, basestring):
            result.extend(flatten(el))
        else:
            result.append(el)
    return result

all_urls = flatten(urls.values())
m = 50 # max numer of urls per query else we do multiple queries

print('=== Triple click text to highlight entire line === \n')

if REGION == 'ALL':
    for i in range( len(all_urls) /  m + 1): 
        n = i*m
        query_string = CHEM + ' site:' + all_urls[n] + ' OR site:'.join(all_urls[n+1:n+m-1])
        print('Search ' + str(i+1) + ': ' + query_string + '\n')
else:
    for i in range( len(urls[REGION]) /  m + 1):
        n = i*m
        query_string = CHEM  + ' site:' + urls[REGION][n] + ' OR site:'.join(urls[REGION][n+1:n+m-1])
        print('Search ' + str(i+1) + ': ' + query_string + '\n')


