web_panels.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698
  1. from flask import flash, render_template, Blueprint, current_app, redirect, request, url_for, session, g, send_from_directory
  2. from datetime import timezone
  3. from ownchatbot.db import get_db, reread_goals, reread_votes, rem_vote, reset_vote, reset_goal, clear_fulfilled_rewards, clear_reward_queue, rem_cool, rem_from_queue
  4. from ownchatbot.reward_handlers import all_active_votes, all_active_goals, all_active_rewards, get_queue, fulfill_reward, save_rewards, activate_category, deactivate_category, refund_reward, reread_categories, save_config, save_alerts, del_alert_file
  5. from ownchatbot.user_handlers import get_all_users, get_all_users_by_name, refund_points, adjust_points, change_email, get_email_code, del_email_code, save_todolist
  6. from ownchatbot.bot_messages import save_announce
  7. from ownchatbot.owncast_com import send_private_chat
  8. import json
  9. import emoji
  10. from ownchatbot.kofi_handlers import save_kofi_settings, kofi_pngs
  11. import random
  12. import pkce
  13. import requests
  14. from functools import wraps
  15. import os
  16. ocb = Blueprint('web_panels', __name__)
  17. state_value = ''
  18. def requires_login(f):
  19. @wraps(f)
  20. def decorated_function(*args, **kwargs):
  21. if 'user' not in session:
  22. return redirect(url_for('web_panels.login'))
  23. return f(*args, **kwargs)
  24. return decorated_function
  25. @ocb.route('/login') # Remove this comment
  26. def login(): # Verify the streamer using indieauth, to their owncast instance
  27. code_verifier, code_challenge = pkce.generate_pkce_pair() # Generate a code verifier and code challenge
  28. global state_value
  29. state_value = code_verifier
  30. owncast_url = current_app.config['OWNCAST_URL']
  31. client_id = current_app.config['ACCESS_ID']
  32. redirect_url = f'{owncast_url}/api/auth/provider/indieauth?client_id={client_id}&redirect_uri={url_for("web_panels.auth_response", _external=True)}&response_type=code&code_challenge_method=S256&code_challenge={code_challenge}&state={code_verifier}'
  33. return redirect(redirect_url)
  34. @ocb.route('/auth_response')
  35. def auth_response():
  36. code = request.args.get('code')
  37. state = request.args.get('state')
  38. if state == state_value: # Check that the state value returned matches the state value sent
  39. current_app.logger.info(f'CSRF code is valid.')
  40. owncast_url = current_app.config['OWNCAST_URL']
  41. owncast_auth_url = f'{owncast_url}/api/auth/provider/indieauth'
  42. access_id = current_app.config['ACCESS_ID']
  43. access_token = current_app.config['ACCESS_TOKEN']
  44. # https://owncast.online/api/latest/#tag/Auth
  45. # https://aaronparecki.com/2021/04/13/26/indieauth
  46. token_response = requests.post(owncast_auth_url, data={
  47. 'client_id': access_id,
  48. 'client_secret': access_token,
  49. 'code': code,
  50. 'redirect_uri': url_for("web_panels.auth_response", _external=True),
  51. 'grant_type': 'authorization_code',
  52. 'code_verifier': state
  53. })
  54. return_data = token_response.json()
  55. f_return_data = json.dumps(return_data, indent=4)
  56. auth_photo = return_data['profile']['photo']
  57. session['user'] = f_return_data
  58. current_app.logger.info(f'Authenticated.')
  59. return redirect(url_for('web_panels.mgmt'))
  60. else:
  61. current_app.logger.info(f'Invalid CSRF Code.')
  62. return 'Not Authorized'
  63. @ocb.route('/logout')
  64. def logout():
  65. session.pop('user', None)
  66. return redirect(url_for('web_panels.user_panel'))
  67. @ocb.route('/mgmt', methods=['GET']) # The streamer's management panel
  68. @requires_login
  69. def mgmt():
  70. owncast_url = current_app.config['OWNCAST_URL']
  71. db = get_db()
  72. users = get_all_users(db)
  73. utc_timezone = timezone.utc
  74. rewards = current_app.config['REWARDS']
  75. active_rewards = []
  76. for each_reward in all_active_rewards(): # Get the name of all active rewards
  77. active_rewards.append(each_reward)
  78. active_categories = current_app.config['ACTIVE_CAT']
  79. inactive_categories = current_app.config['INACTIVE_CAT']
  80. all_cats = current_app.config['ALL_CAT']
  81. points_interval = current_app.config['POINTS_INTERVAL']
  82. points_award = current_app.config['POINTS_AWARD']
  83. gunicorn_logging = current_app.config['GUNICORN']
  84. prefix = current_app.config['PREFIX']
  85. access_id = current_app.config['ACCESS_ID']
  86. access_token = current_app.config['ACCESS_TOKEN']
  87. kofi_token = current_app.config['KOFI_TOKEN']
  88. kofi_integration = current_app.config['KOFI_INTEGRATION']
  89. kofi_logos = kofi_pngs()
  90. announce_enable = current_app.config['ANNOUNCE_ENABLE']
  91. announce_interval = current_app.config['ANNOUNCE_INTERVAL']
  92. announcements = current_app.config['ANNOUNCEMENTS']
  93. todolist_items = current_app.config['LIST']
  94. active_tab = request.args.get('activeTab')
  95. alerts_dict = current_app.config['ALERTS']
  96. settings_info = [
  97. access_id,
  98. points_interval,
  99. points_award,
  100. gunicorn_logging,
  101. prefix,
  102. access_token,
  103. owncast_url,
  104. kofi_token,
  105. kofi_integration,
  106. announce_enable,
  107. announce_interval
  108. ]
  109. return render_template('mgmt.html',
  110. queue=get_queue(db),
  111. votes=all_active_votes(db),
  112. goals=all_active_goals(db),
  113. rewards=rewards,
  114. active_rewards=active_rewards,
  115. prefix=current_app.config['PREFIX'],
  116. kofi_settings=current_app.config['KOFI_SETTINGS'],
  117. kofi_integration=kofi_integration,
  118. kofi_logos=kofi_logos,
  119. announcements=announcements,
  120. users=users,
  121. utc_timezone=utc_timezone,
  122. active_categories=active_categories,
  123. inactive_categories=inactive_categories,
  124. settings_info=settings_info,
  125. items=todolist_items,
  126. alerts_dict=alerts_dict,
  127. activeTab=active_tab)
  128. @ocb.route('/mgmt_queue', methods=['GET']) # The streamer's management panel
  129. @requires_login
  130. def mgmtqueue():
  131. owncast_url = current_app.config['OWNCAST_URL']
  132. db = get_db()
  133. users = get_all_users(db)
  134. utc_timezone = timezone.utc
  135. rewards = current_app.config['REWARDS']
  136. active_rewards = []
  137. for each_reward in all_active_rewards(): # Get the name of all active rewards
  138. active_rewards.append(each_reward)
  139. return render_template('queue.html',
  140. queue=get_queue(db),
  141. votes=all_active_votes(db),
  142. goals=all_active_goals(db),
  143. rewards=rewards,
  144. utc_timezone=utc_timezone)
  145. @ocb.route('/userpanel', methods=['GET']) # The viewers panel
  146. def user_panel():
  147. db = get_db()
  148. instance = request.args.get('instance')
  149. all_rewards = rewards = current_app.config['REWARDS']
  150. username = request.args.get('username')
  151. points_interval = current_app.config['POINTS_INTERVAL']
  152. points_award = current_app.config['POINTS_AWARD']
  153. if username is not None:
  154. users = get_all_users_by_name(db, username)
  155. else:
  156. users = []
  157. utc_timezone = timezone.utc
  158. return render_template('userpanel.html',
  159. queue=get_queue(db),
  160. votes=all_active_votes(db),
  161. goals=all_active_goals(db),
  162. rewards=all_active_rewards(),
  163. all_rewards=all_rewards,
  164. prefix=current_app.config['PREFIX'],
  165. kofi_settings=current_app.config['KOFI_SETTINGS'],
  166. kofi_integration=current_app.config['KOFI_INTEGRATION'],
  167. points_interval=points_interval,
  168. points_award=points_award,
  169. username=username,
  170. users=users,
  171. instance=instance,
  172. utc_timezone=utc_timezone)
  173. @ocb.route('/mgmt/fulfill', methods=['GET'])
  174. @requires_login
  175. def fulfilled():
  176. db = get_db()
  177. reward_id = request.args.get('reward_id')
  178. username = request.args.get('username')
  179. fulfill_reward(db, reward_id)
  180. return redirect(url_for('web_panels.mgmtqueue'))
  181. @ocb.route('/mgmt/refund', methods=['GET'])
  182. @requires_login
  183. def refund():
  184. db = get_db()
  185. reward_id = request.args.get('reward_id')
  186. reward = request.args.get('reward')
  187. rewards = current_app.config['REWARDS']
  188. points = rewards[reward]['price']
  189. username = request.args.get('username')
  190. user_id = request.args.get('rewarder_id')
  191. refund_points(db, user_id, points) # resets points
  192. refund_reward(db, reward_id) # marks the reward as refunded
  193. return redirect(url_for('web_panels.mgmtqueue'))
  194. @ocb.route('/mgmt/edit_account/<user_id>', methods=['GET', 'POST']) # Streamer manually edit user's account
  195. @requires_login
  196. def edit_account(user_id):
  197. db = get_db()
  198. name = request.args.get('name')
  199. points = request.args.get('points')
  200. email = request.args.get('email')
  201. if request.method == 'POST':
  202. user_id = request.form['user_id']
  203. name = request.form['name']
  204. newpoints = request.form['newpoints']
  205. adjust_points(db, user_id, newpoints)
  206. newemail = request.form['newemail']
  207. if newemail == 'None':
  208. current_app.logger.info(f'No email change requested')
  209. else:
  210. if change_email(db, user_id, newemail):
  211. if newemail == '':
  212. current_app.logger.info(f'Removed {name}\'s email')
  213. else:
  214. current_app.logger.info(f'Changed {name}\'s email to {newemail}')
  215. return redirect(url_for('web_panels.mgmt', activeTab='accounts'))
  216. return render_template('edit_account.html',
  217. name=name,
  218. user_id=user_id,
  219. points=points,
  220. email=email)
  221. @ocb.route('/mgmt/delete/<reward_name>', methods=['GET', 'POST'])
  222. @requires_login
  223. def delete(reward_name):
  224. del_reward = current_app.config['REWARDS']
  225. del_reward.pop(reward_name)
  226. if save_rewards(del_reward):
  227. if rem_cool(reward_name):
  228. rem_from_queue(reward_name)
  229. if reread_votes():
  230. if reread_goals():
  231. pass
  232. return redirect(url_for('web_panels.mgmt', activeTab='managerewards'))
  233. @ocb.route('/mgmt/edit/<reward_name>', methods=['GET', 'POST'])
  234. @requires_login
  235. def edit(reward_name):
  236. active_categories = current_app.config['ACTIVE_CAT']
  237. all_the_rewards = current_app.config['REWARDS']
  238. reward_data = all_the_rewards[reward_name]
  239. all_cats = current_app.config['ALL_CAT']
  240. if request.method == 'POST':
  241. reward_data['cooldown'] = int(request.form['cooldown'])
  242. reward_data['type'] = request.form['type']
  243. if reward_data['type'] == 'goal':
  244. reward_data['target'] = int(request.form['target'])
  245. if "milestones" not in reward_data: # If using old rewards.py, and no milestones key exists, create one
  246. reward_data["milestones"] = {"milestone1": [], "milestone2": [], "milestone3": []}
  247. if request.form['milestone1_points'] == '':
  248. reward_data['milestones']['milestone1'] = []
  249. else:
  250. milestone1_points = int(request.form['milestone1_points'])
  251. reward_data['milestones']['milestone1'] = [request.form['milestone1_desc'], milestone1_points]
  252. if request.form['milestone2_points'] == '':
  253. reward_data['milestones']['milestone2'] = []
  254. else:
  255. milestone2_points = int(request.form['milestone2_points'])
  256. reward_data['milestones']['milestone2'] = [request.form['milestone2_desc'], milestone2_points]
  257. if request.form['milestone3_points'] == '':
  258. reward_data['milestones']['milestone3'] = []
  259. else:
  260. milestone3_points = int(request.form['milestone3_points'])
  261. reward_data['milestones']['milestone3'] = [request.form['milestone3_desc'], milestone3_points]
  262. else:
  263. reward_data['price'] = int(request.form['price'])
  264. reward_data['info'] = emoji.demojize(request.form['info'])
  265. if reward_data['type'] == 'special':
  266. reward_data['cmd'] = request.form['cmd']
  267. reward_data['categories'] = request.form.getlist('category')
  268. reward_data['cooldown'] = int(request.form['cooldown'])
  269. all_the_rewards[reward_name] = reward_data
  270. save_rewards(all_the_rewards)
  271. if reward_data['type'] == 'goal': # Sync goals and votes in the db with rewards.py
  272. reread_goals()
  273. if reward_data['type'] == 'vote':
  274. reread_votes()
  275. return redirect(url_for('web_panels.mgmt', activeTab='managerewards'))
  276. return render_template('edit.html',
  277. all_cats=all_cats,
  278. reward_name=reward_name,
  279. active_categories=active_categories,
  280. reward_data=reward_data)
  281. @ocb.route('/mgmt/settings', methods=['GET', 'POST']) # OwnchatBot settings panel
  282. @requires_login
  283. def settings():
  284. todolist_items = current_app.config['LIST']
  285. points_interval = int(request.form['points_interval'])
  286. points_award = int(request.form['points_award'])
  287. gunicorn_logging = 'gunicorn_logging' in request.form
  288. prefix = request.form['prefix']
  289. access_id = request.form['access_id']
  290. access_token = request.form['access_token']
  291. owncast_url = request.form['owncast_url']
  292. kofi_integration = 'kofi_integration' in request.form
  293. kofi_token = request.form['kofi_token']
  294. config_dict = {
  295. 'POINTS_INTERVAL': points_interval,
  296. 'POINTS_AWARD': points_award,
  297. 'GUNICORN': gunicorn_logging,
  298. 'PREFIX': prefix,
  299. 'ACCESS_ID': access_id,
  300. 'ACCESS_TOKEN': access_token,
  301. 'OWNCAST_URL': owncast_url,
  302. 'KOFI_TOKEN': kofi_token,
  303. 'KOFI_INTEGRATION': kofi_integration
  304. }
  305. if save_config(config_dict): # Save new config.py
  306. current_app.logger.info('Saved new config.')
  307. return redirect(url_for('web_panels.mgmt', activeTab='settings'))
  308. @ocb.route('/mgmt/announcements', methods=['GET', 'POST']) # OwnchatBot settings panel
  309. @requires_login
  310. def announcements():
  311. announce_enable = 'announce_enable' in request.form
  312. announce_interval = int(request.form['announce_interval'])
  313. new_announcements = []
  314. new_announcements = request.form['announcements'].strip().split('\n')
  315. announce_dict = {
  316. 'ANNOUNCEMENTS': new_announcements,
  317. 'ANNOUNCE_ENABLE': announce_enable,
  318. 'ANNOUNCE_INTERVAL': announce_interval
  319. }
  320. if save_announce(announce_dict): # Save new announce.py
  321. current_app.logger.info('Saved new announcements.')
  322. return redirect(url_for('web_panels.mgmt', activeTab='announcements'))
  323. @ocb.route('/mgmt/ksettings', methods=['GET', 'POST']) # OwnchatBot settings panel
  324. @requires_login
  325. def ksettings():
  326. kofi_settings_dict = current_app.config['KOFI_SETTINGS']
  327. if request.method == 'POST':
  328. enable_donations = 'enable_donations' in request.form
  329. set_donation_points = request.form['set_donation_points']
  330. enable_subs = 'enable_subs' in request.form
  331. sub_points = int(request.form['sub_points'])
  332. kofi_url = request.form['kofi_url']
  333. kofi_logo = request.form.get('kofi_logo')
  334. kofi_settings_dict['donations'] = enable_donations
  335. kofi_settings_dict['subs'] = enable_subs
  336. kofi_settings_dict['sub_points'] = sub_points
  337. kofi_settings_dict['kofi_url'] = kofi_url
  338. kofi_settings_dict['kofi_logo'] = kofi_logo
  339. if save_kofi_settings(kofi_settings_dict):
  340. current_app.logger.info(f'Saved Kofi settings')
  341. return redirect(url_for('web_panels.mgmt', activeTab='kofi-settings'))
  342. @ocb.route('/mgmt/add/<reward_type>', methods=['GET', 'POST'])
  343. @requires_login
  344. def add(reward_type):
  345. all_cats = current_app.config['ALL_CAT']
  346. active_categories = current_app.config['ACTIVE_CAT']
  347. all_the_rewards = current_app.config['REWARDS']
  348. if request.method == 'POST':
  349. name = request.form['name']
  350. name = name.lower() # Force the name to all lower case
  351. name = emoji.demojize(name) # Remove any emojis
  352. name = name.replace(" ", "") # Remove any spaces from the name
  353. type = request.form['type']
  354. if name in all_the_rewards: # Check for duplicate reward names
  355. flash("A reward with this name already exists.", "error") # Flash an error message
  356. return redirect(url_for('web_panels.add', reward_type=reward_type)) # Redirect back to the add page
  357. if type != 'category': # If we're only adding a category, skip all of this
  358. cooldown = int(request.form['cooldown'])
  359. if type == 'redeem' or type == 'special' or type == 'vote':
  360. price = int(request.form['price'])
  361. if type == 'goal':
  362. target = int(request.form['target'])
  363. milestone1_desc = request.form['milestone1_desc']
  364. if request.form['milestone1_points'] == '':
  365. milestone1_points = ''
  366. else:
  367. milestone1_points = int(request.form['milestone1_points'])
  368. milestone2_desc = request.form['milestone2_desc']
  369. if request.form['milestone2_points'] == '':
  370. milestone2_points = ''
  371. else:
  372. milestone2_points = int(request.form['milestone2_points'])
  373. milestone3_desc = request.form['milestone3_desc']
  374. if request.form['milestone3_points'] == '':
  375. milestone3_points = ''
  376. else:
  377. milestone3_points = int(request.form['milestone3_points'])
  378. info = request.form['info']
  379. info = emoji.demojize(info) # Remove any emojis
  380. if type == 'special':
  381. cmd = request.form['cmd']
  382. categories = request.form.getlist('category')
  383. if type == 'redeem':
  384. if categories == ['']:
  385. all_the_rewards[name] = {'price': price, 'type': type, 'info': info, 'cooldown': cooldown}
  386. else:
  387. all_the_rewards[name] = {'price': price, 'type': type, 'info': info, 'categories': categories, 'cooldown': cooldown}
  388. if type == 'goal':
  389. if categories == ['']:
  390. all_the_rewards[name] = {'target': target, 'type': type, 'info': info, 'cooldown': cooldown}
  391. else:
  392. all_the_rewards[name] = {'target': target, 'type': type, 'info': info, 'categories': categories, 'cooldown': cooldown}
  393. all_the_rewards[name]["milestones"] = {"milestone1": [], "milestone2": [], "milestone3": []} # Create empty milestones key
  394. if milestone1_points:
  395. all_the_rewards[name]["milestones"]["milestone1"] = [milestone1_desc, milestone1_points]
  396. if milestone2_points:
  397. all_the_rewards[name]["milestones"]["milestone2"] = [milestone2_desc, milestone2_points]
  398. if milestone3_points:
  399. all_the_rewards[name]["milestones"]["milestone3"] = [milestone3_desc, milestone3_points]
  400. if type == 'vote':
  401. if categories == ['']:
  402. all_the_rewards[name] = {'price': price, 'type': type, 'info': info}
  403. else:
  404. all_the_rewards[name] = {'price': price, 'type': type, 'info': info, 'categories': categories, 'cooldown': cooldown}
  405. if type == 'special':
  406. if categories == ['']:
  407. all_the_rewards[name] = {'price': price, 'type': type, 'info': info, 'cmd': cmd, 'cooldown': cooldown}
  408. else:
  409. all_the_rewards[name] = {'price': price, 'type': type, 'info': info, 'cmd': cmd, 'categories': categories, 'cooldown': cooldown}
  410. save_rewards(all_the_rewards)
  411. if type == 'goal': # Remove old goals and votes from the database
  412. reread_goals()
  413. if type == 'vote':
  414. reread_votes()
  415. else: # If we're only adding a category
  416. inactive_categories = current_app.config['INACTIVE_CAT']
  417. inactive_categories.append(name) # Add it to the INACTIVE_CAT variable
  418. reread_categories() # Write it to categories.py
  419. return redirect(url_for('web_panels.mgmt', activeTab='managerewards'))
  420. return render_template('add.html',
  421. all_cats=all_cats,
  422. reward_type=reward_type,
  423. active_categories=active_categories)
  424. @ocb.route('/set_viewer_email', methods=['GET', 'POST'])
  425. def set_viewer_email():
  426. db = get_db()
  427. mail_reg_code = int(request.form['code'])
  428. user_id = request.form['user_id']
  429. db_mail_reg_code = get_email_code(db, user_id)
  430. new_email = request.form['new_email']
  431. instance = request.form['instance']
  432. user_name = request.form['user_name']
  433. if mail_reg_code == db_mail_reg_code:
  434. if change_email(db, user_id, new_email):
  435. del_email_code(db, user_id)
  436. flash(f"Email Address \"{new_email}\" successfully registered.", "success")
  437. send_private_chat(user_id, f'{user_name}, thanks for registering for Kofi perks! I appreciate your support!')
  438. current_app.logger.info(f'Changed {user_id}\'s email to {new_email}')
  439. else:
  440. flash(f"Incorrect code. Email Address \"{new_email}\" was not registered.", "failure")
  441. current_app.logger.info(f'The code entered, \"{mail_reg_code}\", does not match \"{db_mail_reg_code}\" found in database.')
  442. return redirect(url_for('web_panels.user_panel', instance=instance, username=user_name))
  443. @ocb.route('/mgmt/activate/<category>', methods=['GET', 'POST'])
  444. @requires_login
  445. def activate(category):
  446. activate_category(category)
  447. return redirect(url_for('web_panels.mgmt', activeTab='categories'))
  448. @ocb.route('/mgmt/deactivate/<category>', methods=['GET', 'POST'])
  449. @requires_login
  450. def deactivate(category):
  451. deactivate_category(category)
  452. return redirect(url_for('web_panels.mgmt', activeTab='categories'))
  453. @ocb.route('/mgmt/delcat/<cat_name>/<cat_act>', methods=['GET', 'POST'])
  454. @requires_login
  455. def delcat(cat_name, cat_act):
  456. active_categories = current_app.config['ACTIVE_CAT']
  457. inactive_categories = current_app.config['INACTIVE_CAT']
  458. if cat_act == 'inactive':
  459. inactive_categories.remove(cat_name)
  460. else:
  461. active_categories.remove(cat_name)
  462. reread_categories()
  463. current_rewards = current_app.config['REWARDS']
  464. for reward, details in current_rewards.items(): # Remove from rewards.py as well
  465. if cat_name in details['categories']:
  466. details['categories'].remove(cat_name)
  467. save_rewards(current_rewards)
  468. return redirect(url_for('web_panels.mgmt', activeTab='categories'))
  469. @ocb.route('/mgmt/reset/<reward_name>/<reward_type>', methods=['GET', 'POST']) # Reset votes and goals to zero
  470. @requires_login
  471. def reset(reward_name, reward_type):
  472. if reward_type == "goal":
  473. reset_goal(reward_name)
  474. if reward_type == "vote":
  475. reset_vote(reward_name)
  476. return redirect(url_for('web_panels.mgmt', activeTab='managerewards'))
  477. @ocb.route('/mgmt/rereadvotes', methods=['GET', 'POST'])
  478. def rereadv():
  479. reread_votes()
  480. return redirect(url_for('web_panels.mgmt'))
  481. @ocb.route('/mgmt/clearfulfilled', methods=['GET', 'POST'])
  482. @requires_login
  483. def clearfulfilled():
  484. if clear_fulfilled_rewards():
  485. current_app.logger.info('Cleared fulfilled rewards.')
  486. return redirect(url_for('web_panels.mgmtqueue'))
  487. @ocb.route('/mgmt/clearqueue', methods=['GET', 'POST'])
  488. @requires_login
  489. def clear_queue():
  490. clear_reward_queue()
  491. return redirect(url_for('web_panels.mgmtqueue'))
  492. @ocb.route('/mgmt/addtodoitem', methods=['POST'])
  493. @requires_login
  494. def add_todo_item():
  495. if request.method == 'POST':
  496. todolist_items = current_app.config['LIST']
  497. item = request.form.get('item')
  498. if item:
  499. todolist_items.append({'name': item, 'crossed': 'no'})
  500. if save_todolist(todolist_items): # Save todo list
  501. current_app.logger.info('Saved to-do list.')
  502. return redirect(url_for('web_panels.mgmt', activeTab='todolist'))
  503. return redirect(url_for('web_panels.mgmt', activeTab='todolist'))
  504. @ocb.route('/mgmt/cross/<int:item_id>')
  505. @requires_login
  506. def cross(item_id):
  507. todolist_items = current_app.config['LIST']
  508. if 0 <= item_id < len(todolist_items): # Make sure the item exists
  509. todolist_items[item_id]['crossed'] = 'yes'
  510. if save_todolist(todolist_items): # Save todo list
  511. current_app.logger.info('Saved to-do list.')
  512. return redirect(url_for('web_panels.mgmt', activeTab='todolist'))
  513. @ocb.route('/mgmt/uncross/<int:item_id>')
  514. @requires_login
  515. def uncross(item_id):
  516. todolist_items = current_app.config['LIST']
  517. if 0 <= item_id < len(todolist_items):
  518. todolist_items[item_id]['crossed'] = 'no'
  519. if save_todolist(todolist_items): # Save todo list
  520. current_app.logger.info('Saved to-do list.')
  521. return redirect(url_for('web_panels.mgmt', activeTab='todolist'))
  522. @ocb.route('/mgmt/remtodoitem/<int:item_id>')
  523. @requires_login
  524. def rem_todo_item(item_id):
  525. todolist_items = current_app.config['LIST']
  526. if 0 <= item_id < len(todolist_items): # Make sure the item exists
  527. removed = todolist_items.pop(item_id)
  528. current_app.logger.info(f'Removed \"{removed}\" from the to-do list.')
  529. if save_todolist(todolist_items): # Save todo list
  530. current_app.logger.info('Saved to-do list.')
  531. return redirect(url_for('web_panels.mgmt', activeTab='todolist'))
  532. @ocb.route('/mgmt/clearlist')
  533. @requires_login
  534. def clear_list():
  535. todolist_items = current_app.config['LIST']
  536. todolist_items = [] # Clear the list
  537. if save_todolist(todolist_items): # Save todo list
  538. current_app.logger.info('Saved to-do list.')
  539. return redirect(url_for('web_panels.mgmt', activeTab='todolist'))
  540. @ocb.route('/mgmt/alertupload/<alert_type>', methods=['POST'])
  541. @requires_login
  542. def alert_upload(alert_type):
  543. alerts_dict = current_app.config['ALERTS']
  544. alert_file = request.files[alert_type]
  545. if alert_type in alerts_dict.keys(): # If the alert already exists, delete the existing alert file
  546. old_file = alerts_dict[alert_type.upper()]
  547. current_app.logger.info(f'{alert_type} already set.')
  548. if del_alert_file(old_file):
  549. pass
  550. filepath = os.path.join(current_app.config['ASSETS_FOLDER'], alert_file.filename)
  551. alert_file.save(filepath)
  552. alerts_dict[alert_type] = alert_file.filename
  553. if save_alerts(alerts_dict): # Save new alerts.py
  554. return redirect(url_for('web_panels.mgmt', activeTab='alerts'))
  555. return redirect(url_for('web_panels.mgmt', activeTab='alerts'))
  556. @ocb.route('/mgmt/delalert/<alert_type>')
  557. @requires_login
  558. def del_alert(alert_type):
  559. alerts_dict = current_app.config['ALERTS']
  560. try:
  561. alert_file = alerts_dict[alert_type]
  562. alerts_dict.pop(alert_type)
  563. if save_alerts(alerts_dict):
  564. current_app.logger.info(f'Removed {alert_type} from alerts.py.')
  565. if del_alert_file(alert_file):
  566. pass
  567. except KeyError:
  568. current_app.logger.info(f'No {alert_type} alert set. Nothing to do.')
  569. return redirect(url_for('web_panels.mgmt', activeTab='alerts'))
  570. @ocb.route('/alerts/<alert_name>') # Route for displaying alert images
  571. def alerts(alert_name):
  572. return send_from_directory(current_app.config['ASSETS_FOLDER'], alert_name)
  573. @ocb.route('/alert/<alert_type>') # Route for alerts overlay
  574. def ocb_alert(alert_type):
  575. alert_type = alert_type.upper()+'_ALERT'
  576. alerts_dict = current_app.config['ALERTS']
  577. try:
  578. alert_name = alerts_dict[alert_type]
  579. except KeyError:
  580. current_app.logger.info(f'No {alert_type} alert set. Nothing to do.')
  581. return f'You have not configured alert type \"{alert_type}\"'
  582. if alert_type == 'FOLLOWER_ALERT':
  583. return render_template('follower.html',
  584. alert_name=alert_name)
  585. elif alert_type == 'MILESTONE_ALERT':
  586. return render_template('rmilestone.html',
  587. alert_name=alert_name)
  588. elif alert_type == 'GOAL_ALERT':
  589. return render_template('rgoal.html',
  590. alert_name=alert_name)
  591. @ocb.route('/goals', methods=['GET']) # Route for goals overlay
  592. def goals():
  593. db = get_db()
  594. return render_template('goals.html',
  595. goals=all_active_goals(db),
  596. rewards=all_active_rewards())
  597. @ocb.route('/votes', methods=['GET']) # Route for votes overlay
  598. def votes():
  599. db = get_db()
  600. return render_template('votes.html',
  601. votes=all_active_votes(db))
  602. @ocb.route('/todo')
  603. def todo():
  604. todolist_items = current_app.config['LIST']
  605. return render_template('list.html', items=todolist_items)