import re

geos = [
    'US', 'FR', 'SE', 'MX', 'BR', 'DE', 'GB', 'IN', 'CA', 'ES',
    'IT', 'AR', 'AU', 'CL', 'CO', 'NL', 'PH', 'CH', 'ZA',
    'IE', 'NZ', 'PT', 'TW', 'JM', 'PE', 'CR', 'BS', 'HN',
    'GD', 'EC', 'DO', 'SV', 'UY', 'PA', 'HT', 'MS', 'BO', 'BM',
    'PY', 'DM', 'VE', 'GT', 'DK', 'NO', 'SA', 'PL', 'BE', 'EG',
    'ID', 'TH', 'MY', 'SG', 'VN', 'HK', 'AT', 'TR', 'GR',
]

small_geos = [
    'CA', 'ES', 'IT', 'AR', 'AU', 'CL', 'CO', 'NL', 'PE', 'CH',
    'ZA', 'IE', 'NZ', 'PT', 'TW', 'JM', 'CR', 'BS', 'HN',
    'GD', 'EC', 'DO', 'SV', 'UY', 'PA', 'HT', 'MS', 'BO', 'BM',
    'PY', 'DM', 'VE', 'GT', 'SE', 'SA', 'PL', 'BE', 'EG',
    'ID', 'TH', 'MY', 'SG', 'VN', 'HK', 'AT', 'TR', 'GR',
]

smaller_geos = [
    'CH', 'ZA', 'IE', 'NZ', 'PT', 'TW', 'JM', 'CR', 'BS', 'HN',
    'GD', 'EC', 'DO', 'SV', 'UY', 'PA', 'HT', 'MS', 'BO', 'BM',
    'PY', 'DM', 'VE', 'GT', 'DK', 'NO', 'SA', 'BE', 'EG',
    'TH', 'MY', 'SG', 'VN', 'HK', 'AT', 'GR',
]

global_geos = [
    'XX'
]
# model_version directory naming convention: t<YYMM>_<descriptor>
# e.g. t2509_snowflake, t2604_noamazon_noyoutube
MODEL_VERSION_REGEX = r't\d{4}_[a-z][a-z0-9_]*'
MODEL_VERSION_RE = re.compile(MODEL_VERSION_REGEX)

REPORT_COUNTRIES = 'countries'
REPORT_GLOBAL = 'global'
REPORT_FINAL_TABLES = {
    REPORT_GLOBAL: 'tadas_global_daily_predictions',
    REPORT_COUNTRIES: 'tadas_daily_predictions'
}
REPORTS = list(REPORT_FINAL_TABLES.keys())