From 3af27f651231488e2b404bcddfaeb2f29d69452c Mon Sep 17 00:00:00 2001 From: Eriks K Date: Wed, 3 Mar 2021 13:15:26 +0200 Subject: [PATCH] Cookie dump migration --- erepublik/citizen.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/erepublik/citizen.py b/erepublik/citizen.py index 1bb9d2d..f76f699 100644 --- a/erepublik/citizen.py +++ b/erepublik/citizen.py @@ -570,8 +570,14 @@ class BaseCitizen(access_points.CitizenAPI): with open(dump_name) as f: data = utils.json.load(f, object_hook=utils.json_decode_object_hook) player = cls(data['config']['email'], "") - for cookie in data['cookies']: - player._req.cookies.set(**cookie) + if data.get('cookies'): + cookies = data.get('cookies') + if isinstance(cookies, list): + for cookie in data['cookies']: + player._req.cookies.set(**cookie) + else: + player._req.cookies.update(cookies) + player._req.headers.update({"User-Agent": data['user_agent']}) for k, v in data.get('config', {}).items(): if hasattr(player.config, k):