Added comments and more options

This commit is contained in:
Jean TOULZA 2018-04-12 15:03:58 +02:00
parent 1b0e42c538
commit 5790a2b60f
2 changed files with 67 additions and 30 deletions

59
auto.py
View File

@ -16,24 +16,21 @@ BASE = "https://scplanner.net/"
COOKIE_NAME = config["CONFIG"]["cookie_name"]
COOKIE_VALUE = config["CONFIG"]["cookie_value"]
print(COOKIE_NAME+"="+COOKIE_VALUE)
def get_cookie_jar():
try:
from cookielib import Cookie, CookieJar # Python 2
except ImportError:
from http.cookiejar import Cookie, CookieJar # Python 3.
from http.cookies import SimpleCookie
'''
Creates a cookie jar for the mechanicalsoup browser
'''
from http.cookiejar import Cookie, CookieJar
cj = CookieJar()
# Cookie(version, name, value, port, port_specified, domain,
# domain_specified, domain_initial_dot, path, path_specified,
# secure, discard, comment, comment_url, rest)
c = Cookie(version=0, name=COOKIE_NAME, value=COOKIE_VALUE, port=None, port_specified=False, domain='scplanner.net',
domain_specified=True, domain_initial_dot=False, path='/', path_specified=True, secure=False, expires=None, discard=True, comment=None, comment_url=None, rest=None, rfc2109=True)
cj.set_cookie(c)
return cj
def resolve(url):
'''
Transforms a 'url' into a soundcloud object. False if not found.
'''
client_id = "nF5U0gNsNB8529U1rHetpIywdIlnEKk7"
client = soundcloud.Client(client_id=client_id)
try:
@ -51,6 +48,9 @@ def resolve(url):
return False
def get_client_id():
'''
Tricky part to get the current public client_id of SoundCloud's website
'''
body = requests.get("https://soundcloud.com").text
reg_app = r'https:\/\/a-v2\.sndcdn\.com\/assets\/app(.*).js'
matches = re.search(reg_app,body)
@ -86,6 +86,10 @@ def auto(track_url):
try:
br.select_form("#autoschForm")
br["urls"] = track_url
use_all_accounts = False
try:
if config["AUTO"]["use_all_accounts"] == "yes":
soup = BeautifulSoup(rep.text, "html.parser")
accounts_select = soup.find("select", {"name": "account[]"})
accounts = []
@ -93,19 +97,21 @@ def auto(track_url):
accounts.append(account_option["value"])
br["account[]"] = tuple(accounts)
#br["account[]"] = (107375074, 195766419)
except:
pass
rep = br.submit_selected()
except:
raise Exception("ERROR 131: There was a problem when posting URL on SCPlanner. Is your account still valid?")
raise Exception("There was a problem when posting URL on SCPlanner. Is your account still valid?")
rep = rep.text
rep_json = json.loads(rep)
if not isinstance(rep_json, list):
if rep_json["title"] == "Error S-788":
raise Exception("You have not set your account preferences. Please set them here: https://scplanner.net/account")
raise Exception("You have probably set use_all_accounts to 'no' and haven't set your account preferences. Please set them here: https://scplanner.net/account")
else:
print(rep_json)
raise Exception("Something weird happened. Please tell Amazed#3330.")
raise Exception("Unknown error. Please fill an issue on https://github.com/jeantoulza/scplanner-discord.")
type = rep_json[0]["type"]
if type == "success" and len(rep_json[0]["success_schedules"]) > 0:
group = rep_json[0]["success_schedules"][0]["group"]
@ -114,12 +120,23 @@ def auto(track_url):
if track:
return text+"\nSee https://scplanner.net/calendar/reposts/{0}/{1}/{2}".format(soundcloud_id, track.id, group)
else:
raise Exception("ERROR 312: Schedule was done but I could not find track :( Please PM Amazed#3330.")
raise Exception("Schedule was done but I could not find track :(")
else:
raise Exception("ERROR 851: Track not found :(")
raise Exception("Track not found, or something worse happened :(")
def has_role(user, role):
'''
Returns True if 'user' has the 'role' role, False otherwise.
'''
for u_role in user.roles:
if u_role.name == role:
return True
return False
def start_bot():
'''
Main bot loop
'''
bot = commands.Bot(command_prefix='!')
@bot.event
async def on_ready():
@ -128,9 +145,17 @@ def start_bot():
print(bot.user.id)
print('------')
@bot.command()
async def repost(url : str):
@bot.command(pass_context=True)
async def repost(ctx, url : str):
"""Reposts an URL on SCPlanner."""
user = ctx.message.author
if not isinstance(user, discord.Member):
await bot.say("Please use this command in a server.")
return
if not has_role(user, config["CONFIG"]["reposters_role"]):
await bot.say("You do no have the required role.")
return
try:
calendar_link = auto(url)
await bot.say(calendar_link)

View File

@ -1,4 +1,16 @@
[CONFIG]
# do not change this
cookie_name = ci_session
# log in on SCPlanner, and grap the "ci_session" cookie value
cookie_value = get_your_own
# get one here: https://discordapp.com/developers/applications/me
bot_token = get_your_own
# reposters_role: only members of this role will be able to repost
reposters_role = reposters
[AUTOSCHEDULE]
# use_all_accounts: "yes" if you want to schedule reposts on all subs, "no" (default) if you want the bot to use your saved preferences
use_all_accounts = no