diff --git a/rowers/idoklad.py b/rowers/idoklad.py index 6c04c45a..d2199c06 100644 --- a/rowers/idoklad.py +++ b/rowers/idoklad.py @@ -1,5 +1,5 @@ import requests -import json +import json, yaml from requests.auth import HTTPBasicAuth import urllib.parse from rowers.utils import dologging @@ -14,7 +14,9 @@ contacts_url = 'https://api.idoklad.cz/v3/Contacts' from rowers.models import iDokladToken -idoklad_countries = json.loads(open('rowers/idoklad_countries.json').read())["Data"]["Items"] +#idoklad_countries = json.loads(open('rowers/idoklad_countries.json').read())["Data"]["Items"] +with open('rowers/idoklad_countries.yaml') as f: + idoklad_countries = yaml.load(f, Loader=yaml.FullLoader)["Data"]["Items"] def get_country_id(code): for c in idoklad_countries: