149 lines
4.3 KiB
Python
Executable file
149 lines
4.3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
from bs4 import BeautifulSoup
|
|
from selenium import webdriver
|
|
from selenium.webdriver.common.keys import Keys
|
|
from selenium.common.exceptions import NoSuchElementException
|
|
import multiprocessing, urllib.request, urllib.error, urllib.parse, time, os
|
|
import dbtools, phototools, setuptools
|
|
|
|
def status(driver):
|
|
if "/main/login.php" not in driver.page_source:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def loadPage(url, driver, period=5,init=False):
|
|
if not (init or status(driver)):
|
|
login(driver)
|
|
driver.get(url)
|
|
time.sleep(period)
|
|
|
|
def loginHandler(driver, user = setuptools.user(), password = setuptools.password()):
|
|
loadPage("https://www.planetromeo.com/",driver,3,True)
|
|
loadPage("https://www.planetromeo.com/main/login.php",driver,3,True)
|
|
|
|
curfield = driver.find_element_by_name("username")
|
|
curfield.send_keys(user)
|
|
|
|
curfield = driver.find_element_by_name("passwort")
|
|
curfield.send_keys(password)
|
|
|
|
curfield.send_keys(Keys.RETURN)
|
|
time.sleep(3)
|
|
|
|
return status(driver)
|
|
|
|
class LoginError(Exception):
|
|
pass
|
|
|
|
def login(driver):
|
|
if not (status(driver) or loginHandler(driver)):
|
|
raise LoginError("Login failed.")
|
|
return True
|
|
|
|
def messageID(url):
|
|
return url.split("=")[-1]
|
|
|
|
def quickshareHandler(driver, url, sender):
|
|
nurl = "https://www.planetromeo.com/" + url if "planetromeo.com" not in url else url
|
|
loadPage(nurl)
|
|
juha = BeautifulSoup(driver.page_source, "html5lib")
|
|
|
|
try:
|
|
links = juha.findAll("a")
|
|
for link in links:
|
|
try:
|
|
purl = "https://www.planetromeo.com/" + link["data-pic"] if "planetromeo.com" not in link["data-pic"] else link["data-pic"]
|
|
phototools.processURL(purl, sender, shutup=True)
|
|
except:
|
|
pass
|
|
except:
|
|
pass
|
|
|
|
def messageHandler(sender, recipient, mid, date, driver, mode = 0, db = dbtools.dbHelper()):
|
|
if mode == 0:
|
|
loadPage("https://www.planetromeo.com/msg/?id=" + mid, driver)
|
|
else:
|
|
loadPage("https://www.planetromeo.com/msg/?type=sent&id=" + mid, driver)
|
|
juha = BeautifulSoup(driver.page_source, "html5lib")
|
|
text = juha.select("div.msg div")[0]
|
|
|
|
db.executeQuery("INSERT INTO messages(id, text, sender_id, recipient_id, created_at) VALUES('%s', '%s', '%s', '%s', '%s');" % (mid, setuptools.unescapeText(text.text).strip(), sender, recipient, date))
|
|
db.commit()
|
|
|
|
try:
|
|
links = juha.findAll("a")
|
|
for link in links:
|
|
if "/pix/popup.php/" in link["href"]:
|
|
phototools.processURL(link["href"], sender)
|
|
try:
|
|
db.executeQuery("INSERT INTO photos(mid, pid) VALUES('%s', '%s');" % (mid, phototools.parseurl(link["href"]).split('/')[-1]))
|
|
except:
|
|
pass
|
|
if "/quickshare/" in link["href"]:
|
|
quickshareHandler(driver, link["href"], sender)
|
|
except:
|
|
pass
|
|
|
|
def pageHandler(driver, db = dbtools.dbHelper(), ret = {"rec": 0,"snt": 0}):
|
|
mode = 0
|
|
abort = True
|
|
juha = BeautifulSoup(driver.page_source, "html5lib")
|
|
|
|
if "sent" in driver.current_url:
|
|
mode = 1
|
|
|
|
try:
|
|
for msg in juha.select("table.messageCenter tr")[1:]:
|
|
try:
|
|
data = msg.findAll('td')
|
|
user = data[1].string
|
|
mid = messageID(data[2].find("a")["href"])
|
|
date = data[3].string
|
|
if not db.checkID(mid):
|
|
abort = False
|
|
if mode == 0:
|
|
messageHandler(user, setuptools.user(), mid, date, driver, mode, db)
|
|
ret["rec"] += 1
|
|
else:
|
|
messageHandler(setuptools.user(), user, mid, date, driver, mode, db)
|
|
ret["snt"] += 1
|
|
except IndexError:
|
|
pass
|
|
except IndexError:
|
|
return False
|
|
|
|
if abort:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
def siteHandler(driver, mode = 0, ret = {"rec": 0,"snt": 0}, p = 0, db = dbtools.dbHelper()):
|
|
if mode == 0:
|
|
loadPage("https://www.planetromeo.com/mitglieder/messages/uebersicht.php?view=all&seite=" + str(p), driver)
|
|
else:
|
|
loadPage("https://www.planetromeo.com/mitglieder/messages/uebersicht.php?view=sent&seite=" + str(p), driver)
|
|
|
|
if pageHandler(driver, db, ret):
|
|
siteHandler(driver, mode, ret, p+1, db)
|
|
|
|
def mainHandler(ret = {"rec": 0,"snt": 0}):
|
|
siteHandler(driver, 0, ret, db=db)
|
|
siteHandler(driver, 1, ret, db=db)
|
|
|
|
if __name__ == "__main__":
|
|
db = dbtools.dbHelper()
|
|
driver = webdriver.Firefox()
|
|
ret = {"rec": 0,"snt": 0}
|
|
if loginHandler(driver):
|
|
p = multiprocessing.Process(target=mainHandler, args=([ret]))
|
|
p.start()
|
|
p.join(1500)
|
|
if p.is_alive():
|
|
p.terminate()
|
|
p.join()
|
|
driver.close()
|
|
print("Received", ret["rec"], "messages")
|
|
print("Sent", ret["snt"], "messages")
|