fix cookie saving

master
Maximilian Zettler 4 years ago
parent dfe7339431
commit 18ef9ac45f
  1. 13
      timebot.py

@ -48,8 +48,9 @@ class TimeBot:
try: try:
self._load_session_cookies(self._session) self._load_session_cookies(self._session)
except FileNotFoundError as e: except FileNotFoundError as e:
self.logger.error(e) logger.warning(e) # file does not exist... ignored
raise except (EOFError, pickle.UnpicklingError) as e:
logger.warning(e) # file seems to be corrupt... ignoring
request = self._session.get(self.baseurl + "Employee/GetEmployeeList") request = self._session.get(self.baseurl + "Employee/GetEmployeeList")
if 400 <= request.status_code < 500: if 400 <= request.status_code < 500:
self._login(self._session) self._login(self._session)
@ -100,8 +101,8 @@ class TimeBot:
:param requests.Session session: the requests session which will be updated with the loaded cookies :param requests.Session session: the requests session which will be updated with the loaded cookies
""" """
with open(".kekse", "rb") as f: if self._save_session:
if self._save_session: with open(".kekse", "rb") as f:
session.cookies.update(requests.utils.cookiejar_from_dict(pickle.load(f))) session.cookies.update(requests.utils.cookiejar_from_dict(pickle.load(f)))
def add_entry(self, punch_datetime: datetime.datetime, entry_code: int, note: str = None) -> requests.Response: def add_entry(self, punch_datetime: datetime.datetime, entry_code: int, note: str = None) -> requests.Response:
@ -274,8 +275,8 @@ if __name__ == '__main__':
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read(args.c) config.read(args.c)
tb = TimeBot(baseurl=config["general"]["baseurl"], user=args.u, password=args.p, ask_for_password=args.save_cookies, tb = TimeBot(baseurl=config["general"]["baseurl"], user=args.u, password=args.p, ask_for_password=True,
save_session=True) save_session=args.save_cookies)
if args.subparser_name == "punch": if args.subparser_name == "punch":
if args.s == "now": if args.s == "now":
punch_datetime = datetime.datetime.now() punch_datetime = datetime.datetime.now()

Loading…
Cancel
Save