fix cookie saving

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

@ -48,8 +48,9 @@ class TimeBot:
try:
self._load_session_cookies(self._session)
except FileNotFoundError as e:
self.logger.error(e)
raise
logger.warning(e) # file does not exist... ignored
except (EOFError, pickle.UnpicklingError) as e:
logger.warning(e) # file seems to be corrupt... ignoring
request = self._session.get(self.baseurl + "Employee/GetEmployeeList")
if 400 <= request.status_code < 500:
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
"""
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)))
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.read(args.c)
tb = TimeBot(baseurl=config["general"]["baseurl"], user=args.u, password=args.p, ask_for_password=args.save_cookies,
save_session=True)
tb = TimeBot(baseurl=config["general"]["baseurl"], user=args.u, password=args.p, ask_for_password=True,
save_session=args.save_cookies)
if args.subparser_name == "punch":
if args.s == "now":
punch_datetime = datetime.datetime.now()

Loading…
Cancel
Save