From 4359974efae9b8ee2114822b8f42759319544b19 Mon Sep 17 00:00:00 2001 From: Sasha MOREL Date: Thu, 1 Dec 2016 11:27:01 +0100 Subject: [PATCH] Check the cookies type Python2: - Added type check for cookies param in HomeSFR class Python3: - Added type check for cookies param in HomeSFR class --- python2/homesfr.py | 4 +++- python3/homesfr.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/python2/homesfr.py b/python2/homesfr.py index 0ff1b5f..26b28e5 100644 --- a/python2/homesfr.py +++ b/python2/homesfr.py @@ -211,8 +211,10 @@ class HomeSFR (Common): self.autologin = False if cookies == None: self.cookies = CookieJar () - else: + elif type (cookies) == CookieJar: self.cookies = cookies + else: + raise TypeError ('Cookies must be CookieJar type.') self.opener = request.build_opener (request.HTTPCookieProcessor (self.cookies)) def __str__ (self): diff --git a/python3/homesfr.py b/python3/homesfr.py index 0040734..4365075 100644 --- a/python3/homesfr.py +++ b/python3/homesfr.py @@ -207,8 +207,10 @@ class HomeSFR (Common): self.autologin = False if cookies == None: self.cookies = CookieJar () - else: + elif type (cookies) == CookieJar: self.cookies = cookies + else: + raise TypeError ('Cookies must be CookieJar type.') self.opener = request.build_opener (request.HTTPCookieProcessor (self.cookies)) def __str__ (self):