150 lines
4.8 KiB
Python
150 lines
4.8 KiB
Python
import requests
|
|
from bs4 import BeautifulSoup
|
|
import mechanize
|
|
import time
|
|
import datetime
|
|
import traceback
|
|
import json
|
|
import http.client
|
|
|
|
with open("config.json", "r") as fp:
|
|
config = json.load(fp)
|
|
|
|
USERNAME = config["username"]
|
|
PASSWORD = config["password"]
|
|
WEBHOOK_URL = config["webhook"]
|
|
LOGOUT = config["logout"]
|
|
BROWSER = mechanize.Browser()
|
|
BROWSER.set_handle_robots(False)
|
|
|
|
|
|
def login():
|
|
login_url = "https://www.giantswd.org/forum/ucp.php?mode=login"
|
|
|
|
# Login
|
|
print("Logging in...")
|
|
BROWSER.open(login_url)
|
|
BROWSER.select_form(id="login")
|
|
BROWSER["username"] = USERNAME
|
|
BROWSER["password"] = PASSWORD
|
|
BROWSER.submit()
|
|
|
|
|
|
def is_logged_in():
|
|
global BROWSER
|
|
response = BROWSER.response()
|
|
if not response:
|
|
return False
|
|
try:
|
|
return b'<a href="./ucp.php" class="header-avatar dropdown-trigger"> <span class="username">'+USERNAME.encode("utf8")+b'</span>' in response.get_data()
|
|
except http.client.IncompleteRead:
|
|
BROWSER = mechanize.Browser()
|
|
return is_logged_in()
|
|
|
|
|
|
ignored = ["AmazedBot"]
|
|
connected = []
|
|
|
|
|
|
def loop():
|
|
global connected
|
|
homepage_url = "https://www.giantswd.org/forum/"
|
|
logout_url = "https://www.giantswd.org/forum/ucp.php?mode=logout"
|
|
newposts_url = "https://www.giantswd.org/forum/search.php?search_id=unreadposts"
|
|
|
|
if not is_logged_in():
|
|
login()
|
|
|
|
if config["logged_users"]:
|
|
# Send a webhook when people connect on GWD forums
|
|
print("Getting connected people...")
|
|
homepage_resp = BROWSER.open(homepage_url)
|
|
homepage_html = homepage_resp.get_data()
|
|
bs = BeautifulSoup(homepage_html, features="html5lib")
|
|
div_online = bs.find("div", attrs={"class": "online-list"})
|
|
div_username_online = div_online.find_all("a", attrs={"class": "username"})
|
|
|
|
new_users = []
|
|
for username_elem in div_username_online:
|
|
user = username_elem.contents[0]
|
|
|
|
if user not in ignored and user not in connected:
|
|
new_users.append(user)
|
|
print("User connected: %s" % user)
|
|
webhook_data = {
|
|
"embeds": [
|
|
{
|
|
"title": "A user has logged in on GWD forums!",
|
|
"description": "**%s** has just logged in" % user,
|
|
"color": 55039,
|
|
"author": {
|
|
"name": "GiantsWD forums"
|
|
},
|
|
"footer": {
|
|
"text": ""
|
|
},
|
|
"timestamp": datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S.%f')[:-3] + 'Z'
|
|
}
|
|
]
|
|
}
|
|
requests.post(WEBHOOK_URL, json=webhook_data)
|
|
|
|
connected = new_users
|
|
|
|
# Get new posts
|
|
print("Getting new posts...")
|
|
has_newposts = False
|
|
newposts_resp = BROWSER.open(newposts_url)
|
|
newposts_html = newposts_resp.get_data()
|
|
bs = BeautifulSoup(newposts_html, features="html5lib")
|
|
all_newposts = bs.find_all("dl", attrs={"class": "topic_unread"})
|
|
for newpost in all_newposts:
|
|
has_newposts = True
|
|
newpost_title = newpost.find(class_="topictitle").contents[0]
|
|
newpost_link = newpost.find(class_="topictitle")["href"]
|
|
newpost_username = newpost.find(class_="username").contents[0]
|
|
newpost_link_real = "https://www.giantswd.org/forum/%s" % (newpost_link[2:])
|
|
|
|
print("%s has created a new post %s (%s)" % (newpost_username, newpost_title, newpost_link_real))
|
|
|
|
webhook_data = {
|
|
"embeds": [
|
|
{
|
|
"title": "A new message has been posted!",
|
|
"description": "**%s** has posted a new message in [**%s**](%s)" % (newpost_username, newpost_title, newpost_link_real),
|
|
"color": 55039,
|
|
"author": {
|
|
"name": "GiantsWD forums"
|
|
},
|
|
"footer": {
|
|
"text": ""
|
|
},
|
|
"timestamp": datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%S.%f')[:-3] + 'Z'
|
|
}
|
|
]
|
|
}
|
|
requests.post(WEBHOOK_URL, json=webhook_data)
|
|
|
|
if has_newposts:
|
|
# Mark all read
|
|
print("Mark all read...")
|
|
a = bs.find("a", class_="mark-read")
|
|
mark_read_url = a["href"]
|
|
BROWSER.open(mark_read_url)
|
|
|
|
# Logout
|
|
if LOGOUT:
|
|
print("Logout...")
|
|
BROWSER.open(logout_url)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
while True:
|
|
try:
|
|
loop()
|
|
except:
|
|
print(datetime.datetime.now())
|
|
traceback.print_exc()
|
|
print("Sleeping...")
|
|
time.sleep(config["check_seconds"])
|