138 lines
4.1 KiB
Python
Executable file
138 lines
4.1 KiB
Python
Executable file
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
'''pokemon.py: Automatically poke back on Facebook'''
|
||
|
||
__author__ = "Klaus-Uwe Mitterer"
|
||
__maintainer__ = __author__
|
||
__email__ = "info@klaus-uwe.me"
|
||
__version__ = "0.3"
|
||
|
||
import urllib, cookielib, urllib2, time, BeautifulSoup, signal, sys, configparser, os.path
|
||
|
||
def getNH(html):
|
||
for input in BeautifulSoup.BeautifulSoup(html).findAll('input'):
|
||
try:
|
||
if input["name"] == "nh":
|
||
return input["value"]
|
||
except:
|
||
pass
|
||
return False
|
||
|
||
def log(text, status=0):
|
||
print text
|
||
|
||
class Account:
|
||
def __init__(self, username, password):
|
||
self.username = username
|
||
self.password = password
|
||
|
||
self.jar = cookielib.MozillaCookieJar("cookies.txt")
|
||
try:
|
||
self.jar.load()
|
||
except:
|
||
pass
|
||
|
||
self.cookie = urllib2.HTTPCookieProcessor(self.jar)
|
||
self.opener = urllib2.build_opener(self.cookie)
|
||
|
||
def login(self):
|
||
|
||
try:
|
||
params = urllib.urlencode({'lsd':'AVoT-4T0','charset_test':'€,´,€,´,水,Д,Є','version':'1','ajax':'0','width':'0','pxr':'0','gps':'0','dimensions':'0','m_ts':int(time.time()),'li':'dflbVIod_lz7592W_lmOrvph','email':self.username,'pass':self.password,'login':'Log+In'})
|
||
req = urllib2.Request('https://m.facebook.com/login.php?refsrc=https%3A%2F%2Fm.facebook.com%2F&refid=8', params)
|
||
res = self.opener.open(req)
|
||
html = res.read()
|
||
|
||
if "Code Generator" in html:
|
||
nh = getNH(html)
|
||
print "You have two-factor authentication enabled. Please enter an approval code (e.g. from your Facebook app)."
|
||
print "Please note: If you get a login error after this, try logging in to Facebook through a browser, confirm the login notification and choose 'Save Browser', then try again."
|
||
appcode = raw_input("Input code: ")
|
||
params = urllib.urlencode({'lsd':'AVrIY3Uu','charset_test':'€,´,€,´,水,Д,Є','approvals_code':appcode,'nh':nh,'codes_submitted':'0', 'submit[Submit+Code]':'Submit+Code'})
|
||
req = urllib2.Request('https://m.facebook.com/login/checkpoint/', params)
|
||
res = self.opener.open(req)
|
||
html = res.read()
|
||
|
||
if "Save Browser" in html:
|
||
nh = getNH(html)
|
||
params = urllib.urlencode({'lsd':'AVrIY3Uu','charset_test':'€,´,€,´,水,Д,Є','name_action_selected':'save_device','nh':nh,'submit[Continue]':'Continue'})
|
||
req = urllib2.Request('https://m.facebook.com/login/checkpoint/', params)
|
||
res = self.opener.open(req)
|
||
html = res.read()
|
||
|
||
except urllib2.HTTPError, e:
|
||
log(e.msg, 1)
|
||
except urllib2.URLError, e:
|
||
log(e.reason[1], 1)
|
||
log("This may occur if you run into Facebook's rate limit.", 1)
|
||
log("Please try again later.", 1)
|
||
return False
|
||
|
||
def fetch(self,url):
|
||
req = urllib2.Request(url,None)
|
||
res = self.opener.open(req)
|
||
content, cookie = res.read(), res.headers.get("Set-Cookie")
|
||
return content, cookie
|
||
|
||
def loggedin(self, webpage=None):
|
||
if webpage == None:
|
||
webpage, session = self.fetch('https://m.facebook.com/')
|
||
if "Logout" in webpage:
|
||
return True
|
||
return False
|
||
|
||
def poke(self):
|
||
webpage, session = self.fetch('https://m.facebook.com/pokes')
|
||
if self.loggedin(webpage):
|
||
links=BeautifulSoup.BeautifulSoup(webpage).findAll('a')
|
||
|
||
for a in links:
|
||
url=a['href']
|
||
if "dom_id_replace" in url:
|
||
if not "suggestion" in url:
|
||
poke = self.fetch('https://m.facebook.com' + url)
|
||
log("Poke!")
|
||
|
||
self.fetch('https://m.facebook.com/notifications.php')
|
||
return True
|
||
return False
|
||
|
||
def loop(self):
|
||
while True:
|
||
try:
|
||
if not self.poke():
|
||
self.login()
|
||
except KeyboardInterrupt:
|
||
self.quit()
|
||
except:
|
||
log("Something unexpected happened... o.O", 1)
|
||
raise
|
||
|
||
def quit(self, status=0):
|
||
log("Exiting...")
|
||
self.jar.save()
|
||
sys.exit(status)
|
||
|
||
def getCredentials(conffile="config.cfg"):
|
||
try:
|
||
config = configparser.RawConfigParser()
|
||
config.read(conffile)
|
||
return config.get("facebook","user"), config.get("facebook","pass")
|
||
except:
|
||
raise IOError("Config file not found. Run setup.py and try again.")
|
||
|
||
if __name__ == "__main__":
|
||
me = Account(*getCredentials())
|
||
me.login()
|
||
|
||
if me.loggedin():
|
||
log("Login successful!")
|
||
else:
|
||
log("Doh!", 1)
|
||
log("Seems like you entered wrong credentials.", 1)
|
||
log("Check the README for details.", 1)
|
||
me.quit(1)
|
||
|
||
me.loop()
|