def login_and_retry(redirect):
_1CH.log('Logging in for url %s' % redirect)
# needed here because login_and_retry was moved to utils. Probably better to pass in but this works and is quick but dirty.
USER_AGENT = ("User-Agent:Mozilla/5.0 (Windows NT 6.2; WOW64)"
"AppleWebKit/537.17 (KHTML, like Gecko)"
"Chrome/24.0.1312.56")
BASE_URL = _1CH.get_setting('domain')
if (_1CH.get_setting("enableDomain")=='true') and (len(_1CH.get_setting("customDomain")) > 10):
BASE_URL = _1CH.get_setting("customDomain")
user = _1CH.get_setting('username')
passwd = _1CH.get_setting('passwd')
url = BASE_URL + '/login.php'
net = Net()
cookiejar = _1CH.get_profile()
cookiejar = os.path.join(cookiejar, 'cookies')
host = re.sub('http://', '', BASE_URL)
headers = {'Referer': redirect, 'Origin': BASE_URL, 'Host': host, 'User-Agent': USER_AGENT}
form_data = {'username': user, 'password': passwd, 'remember': 'on', 'login_submit': 'Login'}
html = net.http_POST(url, headers=headers, form_data=form_data).content
if '<a href="/logout.php">[ Logout ]</a>' in html:
net.save_cookies(cookiejar)
return html
else:
_1CH.log('Failed to login')
print html
def __login(self,redirect):
url = self.base_url + '/login.php'
net = Net()
cookiejar = _1CH.get_profile()
cookiejar = os.path.join(cookiejar, 'cookies')
host = re.sub('http://', '', self.base_url)
headers = {'Referer': redirect, 'Origin': self.base_url, 'Host': host, 'User-Agent': USER_AGENT}
form_data = {'username': self.username, 'password': self.password, 'remember': 'on', 'login_submit': 'Login'}
html = net.http_POST(url, headers=headers, form_data=form_data).content
if '<a href="/logout.php">[ Logout ]</a>' in html:
net.save_cookies(cookiejar)
return True
else:
return False
def __login(self,redirect):
url = self.base_url + '/login.php'
net = Net()
cookiejar = _1CH.get_profile()
cookiejar = os.path.join(cookiejar, 'cookies')
host = re.sub('http://', '', self.base_url)
headers = {'Referer': redirect, 'Origin': self.base_url, 'Host': host, 'User-Agent': USER_AGENT}
form_data = {'username': self.username, 'password': self.password, 'remember': 'on', 'login_submit': 'Login'}
html = net.http_POST(url, headers=headers, form_data=form_data).content
if '<a href="/logout.php"' in html:
net.save_cookies(cookiejar)
return True
else:
return False
def request(self, uri, params=None, query=None, headers=None, return_soup=False, return_json=False):
COOKIE_JAR = vfs.join(COOKIE_PATH, self.service + ".lwp")
net = Net()
net.set_cookies(COOKIE_JAR)
if headers:
headers["Referer"] = self.referrer
headers["Accept"] = self.ACCEPT
headers["User-Agent"] = self.USER_AGENT
else:
headers = {"Referer": self.referrer, "Accept": self.ACCEPT, "User-Agent": self.USER_AGENT}
if query:
uri = uri % urllib.urlencode(query)
if params:
html = net.http_POST(self.base_url + uri, params, headers=headers).content
else:
html = net.http_GET(self.base_url + uri, headers=headers).content
net.save_cookies(COOKIE_JAR)
if return_soup:
return BeautifulSoup(html)
elif return_json:
return json.loads(html)
else:
return html
