Source code for sample_data_utils.geo

import os
from random import choice
from sample_data_utils.utils import memoize

CONTINENTS = (
    ('AF', 'Africa'),
    ('AS', 'Asia'),
    ('EU', 'Europe'),
    ('NA', 'North America'),
    ('SA', 'South America'),
    ('OC', 'Oceania'),
    ('AN', 'Antartica'))


@memoize
[docs]def get_codes():
    """
    >> get_codes()
    ISO	ISO3	ISO-Numeric	fips	Country	Capital	Area(in sq km)	Population	Continent	tld	CurrencyCode	CurrencyName	Phone	Postal Code Format	Postal Code Regex	Languages	geonameid	neighbours	EquivalentFipsCode
    """
    cache_filename = os.path.join(os.path.dirname(__file__), 'data', 'countryInfo.txt')
    data = []
    for line in open(cache_filename, 'r'):
        if not line.startswith('#'):
            data.append(line.split('\t'))

    return data


[docs]def continent():
    return choice(CONTINENTS)


[docs]def country():
    """
    >> country()
    ['SZ', 'SWZ', '748', 'Swaziland', 'AF', '.sz', 'SZL', 'Lilangeni']

    :return: (iso2,iso3,isonum,CountryName,Continent,tld,CurrencyCode,CurrencyName
    """
    codes = get_codes()
    selection = choice(codes)
    return selection[0:3] + selection[4:5] + selection[8:12]


[docs]def iso2():
    codes = get_codes()
    return choice(codes)[0]


[docs]def iso3():
    codes = get_codes()
    return choice(codes)[1]


[docs]def isonum():
    codes = get_codes()
    return choice(codes)[2]


[docs]def lang(*extra):
    # http://www.loc.gov/standards/iso639-2/ISO-639-2_utf-8.txt
    codes = 'aar abk ace ach ada ady afa afh afr ain aka akk alb ale alg alt amh ang anp ' \
            'apa ara arc arg arm arn arp art arw asm ast ath aus ava ave awa aym aze bad ' \
            'bai bak bal bam ban baq bas bat bej bel bem ben ber bho bih bik bin bis bla ' \
            'bnt bos bra bre btk bua bug bul bur byn cad cai car cat cau ceb cel cha chb ' \
            'che chg chi chk chm chn cho chp chr chu chv chy cmc cop cor cos cpe cpf cpp ' \
            'cre crh crp csb cus cze dak dan dar day del den dgr din div doi dra dsb dua ' \
            'dum dut dyu dzo efi egy eka elx eng enm epo est ewe ewo fan fao fat fij fil ' \
            'fin fiu fon fre frm fro frr frs fry ful fur gaa gay gba gem geo ger gez gil ' \
            'gla gle glg glv gmh goh gon gor got grb grc gre grn gsw guj gwi hai hat hau ' \
            'haw heb her hil him hin hit hmn hmo hrv hsb hun hup iba ibo ice ido iii ijo ' \
            'iku ile ilo ina inc ind ine inh ipk ira iro ita jav jbo jpn jpr jrb kaa kab ' \
            'kac kal kam kan kar kas kau kaw kaz kbd kha khi khm kho kik kin kir kmb kok ' \
            'kom kon kor kos kpe krc krl kro kru kua kum kur kut lad lah lam lao lat lav ' \
            'lez lim lin lit lol loz ltz lua lub lug lui lun luo lus mac mad mag mah mai ' \
            'mak mal man mao map mar mas may mdf mdr men mga mic min mis mkh mlg mlt mnc ' \
            'mni mno moh mon mos mul mun mus mwl mwr myn myv nah nai nap nau nav nbl nde ' \
            'ndo nds nep new nia nic niu nno nob nog non nor nqo nso nub nwc nya nym nyn ' \
            'nyo nzi oci oji ori orm osa oss ota oto paa pag pal pam pan pap pau peo per ' \
            'phi phn pli pol pon por pra pro pus qaa-qtz que raj rap rar roa roh rom rum ' \
            'run rup rus sad sag sah sai sal sam san sas sat scn sco sel sem sga sgn shn ' \
            'sid sin sio sit sla slo slv sma sme smi smj smn smo sms sna snd snk sog som ' \
            'son sot spa srd srn srp srr ssa ssw suk sun sus sux swa swe syc syr tah tai ' \
            'tam tat tel tem ter tet tgk tgl tha tib tig tir tiv tkl tlh tli tmh tog ton ' \
            'tpi tsi tsn tso tuk tum tup tur tut tvl twi tyv udm uga uig ukr umb und urd ' \
            'uzb vai ven vie vol vot wak wal war was wel wen wln wol xal xho yao yap yid ' \
            'yor ypk zap zbl zen zgh zha znd zul zun zxx zza'

    return choice(codes.split())