web_panels.py 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697
  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. announce_enable = current_app.config['ANNOUNCE_ENABLE']
  90. announce_interval = current_app.config['ANNOUNCE_INTERVAL']
  91. announcements = current_app.config['ANNOUNCEMENTS']
  92. todolist_items = current_app.config['LIST']
  93. active_tab = request.args.get('activeTab')
  94. alerts_dict = current_app.config['ALERTS']
  95. settings_info = [
  96. access_id,
  97. points_interval,
  98. points_award,
  99. gunicorn_logging,
  100. prefix,
  101. access_token,
  102. owncast_url,
  103. kofi_token,
  104. kofi_integration,
  105. announce_enable,
  106. announce_interval
  107. ]
  108. return render_template('mgmt.html',
  109. queue=get_queue(db),
  110. votes=all_active_votes(db),
  111. goals=all_active_goals(db),
  112. rewards=rewards,
  113. active_rewards=active_rewards,
  114. prefix=current_app.config['PREFIX'],
  115. kofi_settings=current_app.config['KOFI_SETTINGS'],
  116. kofi_integration=kofi_integration,
  117. announcements=announcements,
  118. users=users,
  119. utc_timezone=utc_timezone,
  120. active_categories=active_categories,
  121. inactive_categories=inactive_categories,
  122. settings_info=settings_info,
  123. items=todolist_items,
  124. alerts_dict=alerts_dict,
  125. activeTab=active_tab)
  126. @ocb.route('/mgmt_queue', methods=['GET']) # The streamer's management panel
  127. @requires_login
  128. def mgmtqueue():
  129. owncast_url = current_app.config['OWNCAST_URL']
  130. db = get_db()
  131. users = get_all_users(db)
  132. utc_timezone = timezone.utc
  133. rewards = current_app.config['REWARDS']
  134. active_rewards = []
  135. for each_reward in all_active_rewards(): # Get the name of all active rewards
  136. active_rewards.append(each_reward)
  137. return render_template('queue.html',
  138. queue=get_queue(db),
  139. votes=all_active_votes(db),
  140. goals=all_active_goals(db),
  141. rewards=rewards,
  142. utc_timezone=utc_timezone)
  143. @ocb.route('/userpanel', methods=['GET']) # The viewers panel
  144. def user_panel():
  145. db = get_db()
  146. instance = request.args.get('instance')
  147. all_rewards = rewards = current_app.config['REWARDS']
  148. username = request.args.get('username')
  149. points_interval = current_app.config['POINTS_INTERVAL']
  150. points_award = current_app.config['POINTS_AWARD']
  151. if username is not None:
  152. users = get_all_users_by_name(db, username)
  153. else:
  154. users = []
  155. utc_timezone = timezone.utc
  156. return render_template('userpanel.html',
  157. queue=get_queue(db),
  158. votes=all_active_votes(db),
  159. goals=all_active_goals(db),
  160. rewards=all_active_rewards(),
  161. all_rewards=all_rewards,
  162. prefix=current_app.config['PREFIX'],
  163. kofi_settings=current_app.config['KOFI_SETTINGS'],
  164. kofi_integration=current_app.config['KOFI_INTEGRATION'],
  165. points_interval=points_interval,
  166. points_award=points_award,
  167. username=username,
  168. users=users,
  169. instance=instance,
  170. utc_timezone=utc_timezone)
  171. @ocb.route('/mgmt/fulfill', methods=['GET'])
  172. @requires_login
  173. def fulfilled():
  174. db = get_db()
  175. reward_id = request.args.get('reward_id')
  176. username = request.args.get('username')
  177. fulfill_reward(db, reward_id)
  178. return redirect(url_for('web_panels.mgmtqueue'))
  179. @ocb.route('/mgmt/refund', methods=['GET'])
  180. @requires_login
  181. def refund():
  182. db = get_db()
  183. reward_id = request.args.get('reward_id')
  184. reward = request.args.get('reward')
  185. rewards = current_app.config['REWARDS']
  186. points = rewards[reward]['price']
  187. username = request.args.get('username')
  188. user_id = request.args.get('rewarder_id')
  189. refund_points(db, user_id, points) # resets points
  190. refund_reward(db, reward_id) # marks the reward as refunded
  191. return redirect(url_for('web_panels.mgmtqueue'))
  192. @ocb.route('/mgmt/edit_account/<user_id>', methods=['GET', 'POST']) # Streamer manually edit user's account
  193. @requires_login
  194. def edit_account(user_id):
  195. db = get_db()
  196. name = request.args.get('name')
  197. points = request.args.get('points')
  198. email = request.args.get('email')
  199. if request.method == 'POST':
  200. user_id = request.form['user_id']
  201. name = request.form['name']
  202. newpoints = request.form['newpoints']
  203. adjust_points(db, user_id, newpoints)
  204. newemail = request.form['newemail']
  205. if newemail == 'None':
  206. current_app.logger.info(f'No email change requested')
  207. else:
  208. if change_email(db, user_id, newemail):
  209. if newemail == '':
  210. current_app.logger.info(f'Removed {name}\'s email')
  211. else:
  212. current_app.logger.info(f'Changed {name}\'s email to {newemail}')
  213. return redirect(url_for('web_panels.mgmt', activeTab='accounts'))
  214. return render_template('edit_account.html',
  215. name=name,
  216. user_id=user_id,
  217. points=points,
  218. email=email)
  219. @ocb.route('/mgmt/delete/<reward_name>', methods=['GET', 'POST'])
  220. @requires_login
  221. def delete(reward_name):
  222. del_reward = current_app.config['REWARDS']
  223. del_reward.pop(reward_name)
  224. if save_rewards(del_reward):
  225. if rem_cool(reward_name):
  226. rem_from_queue(reward_name)
  227. if reread_votes():
  228. if reread_goals():
  229. pass
  230. return redirect(url_for('web_panels.mgmt', activeTab='managerewards'))
  231. @ocb.route('/mgmt/edit/<reward_name>', methods=['GET', 'POST'])
  232. @requires_login
  233. def edit(reward_name):
  234. active_categories = current_app.config['ACTIVE_CAT']
  235. all_the_rewards = current_app.config['REWARDS']
  236. reward_data = all_the_rewards[reward_name]
  237. all_cats = current_app.config['ALL_CAT']
  238. if request.method == 'POST':
  239. reward_data['cooldown'] = int(request.form['cooldown'])
  240. reward_data['type'] = request.form['type']
  241. if reward_data['type'] == 'goal':
  242. reward_data['target'] = int(request.form['target'])
  243. if "milestones" not in reward_data: # If using old rewards.py, and no milestones key exists, create one
  244. reward_data["milestones"] = {"milestone1": [], "milestone2": [], "milestone3": []}
  245. if request.form['milestone1_points'] == '':
  246. reward_data['milestones']['milestone1'] = []
  247. else:
  248. milestone1_points = int(request.form['milestone1_points'])
  249. reward_data['milestones']['milestone1'] = [request.form['milestone1_desc'], milestone1_points]
  250. if request.form['milestone2_points'] == '':
  251. reward_data['milestones']['milestone2'] = []
  252. else:
  253. milestone2_points = int(request.form['milestone2_points'])
  254. reward_data['milestones']['milestone2'] = [request.form['milestone2_desc'], milestone2_points]
  255. if request.form['milestone3_points'] == '':
  256. reward_data['milestones']['milestone3'] = []
  257. else:
  258. milestone3_points = int(request.form['milestone3_points'])
  259. reward_data['milestones']['milestone3'] = [request.form['milestone3_desc'], milestone3_points]
  260. else:
  261. reward_data['price'] = int(request.form['price'])
  262. reward_data['info'] = emoji.demojize(request.form['info'])
  263. if reward_data['type'] == 'special':
  264. reward_data['cmd'] = request.form['cmd']
  265. reward_data['categories'] = request.form.getlist('category')
  266. reward_data['cooldown'] = int(request.form['cooldown'])
  267. all_the_rewards[reward_name] = reward_data
  268. save_rewards(all_the_rewards)
  269. if reward_data['type'] == 'goal': # Sync goals and votes in the db with rewards.py
  270. reread_goals()
  271. if reward_data['type'] == 'vote':
  272. reread_votes()
  273. if reward_data['type'] == 'category':
  274. return redirect(url_for('web_panels.mgmt', activeTab='categories'))
  275. else:
  276. return redirect(url_for('web_panels.mgmt', activeTab='managerewards'))
  277. return render_template('edit.html',
  278. all_cats=all_cats,
  279. reward_name=reward_name,
  280. active_categories=active_categories,
  281. reward_data=reward_data)
  282. @ocb.route('/mgmt/settings', methods=['GET', 'POST']) # OwnchatBot settings panel
  283. @requires_login
  284. def settings():
  285. todolist_items = current_app.config['LIST']
  286. points_interval = int(request.form['points_interval'])
  287. points_award = int(request.form['points_award'])
  288. gunicorn_logging = 'gunicorn_logging' in request.form
  289. prefix = request.form['prefix']
  290. access_id = request.form['access_id']
  291. access_token = request.form['access_token']
  292. owncast_url = request.form['owncast_url']
  293. kofi_integration = 'kofi_integration' in request.form
  294. kofi_token = request.form['kofi_token']
  295. config_dict = {
  296. 'POINTS_INTERVAL': points_interval,
  297. 'POINTS_AWARD': points_award,
  298. 'GUNICORN': gunicorn_logging,
  299. 'PREFIX': prefix,
  300. 'ACCESS_ID': access_id,
  301. 'ACCESS_TOKEN': access_token,
  302. 'OWNCAST_URL': owncast_url,
  303. 'KOFI_TOKEN': kofi_token,
  304. 'KOFI_INTEGRATION': kofi_integration
  305. }
  306. if save_config(config_dict): # Save new config.py
  307. current_app.logger.info('Saved new config.')
  308. return redirect(url_for('web_panels.mgmt', activeTab='settings'))
  309. @ocb.route('/mgmt/announcements', methods=['GET', 'POST']) # OwnchatBot settings panel
  310. @requires_login
  311. def announcements():
  312. announce_enable = 'announce_enable' in request.form
  313. announce_interval = int(request.form['announce_interval'])
  314. new_announcements = []
  315. new_announcements = request.form['announcements'].strip().split('\n')
  316. announce_dict = {
  317. 'ANNOUNCEMENTS': new_announcements,
  318. 'ANNOUNCE_ENABLE': announce_enable,
  319. 'ANNOUNCE_INTERVAL': announce_interval
  320. }
  321. if save_announce(announce_dict): # Save new announce.py
  322. current_app.logger.info('Saved new announcements.')
  323. return redirect(url_for('web_panels.mgmt', activeTab='announcements'))
  324. @ocb.route('/mgmt/ksettings', methods=['GET', 'POST']) # OwnchatBot settings panel
  325. @requires_login
  326. def ksettings():
  327. kofi_settings_dict = current_app.config['KOFI_SETTINGS']
  328. if request.method == 'POST':
  329. enable_donations = 'enable_donations' in request.form
  330. set_donation_points = request.form['set_donation_points']
  331. enable_subs = 'enable_subs' in request.form
  332. sub_points = int(request.form['sub_points'])
  333. kofi_url = request.form['kofi_url']
  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. if save_kofi_settings(kofi_settings_dict):
  339. current_app.logger.info(f'Saved Kofi settings')
  340. return redirect(url_for('web_panels.mgmt', activeTab='kofi-settings'))
  341. @ocb.route('/mgmt/add/<reward_type>', methods=['GET', 'POST'])
  342. @requires_login
  343. def add(reward_type):
  344. all_cats = current_app.config['ALL_CAT']
  345. active_categories = current_app.config['ACTIVE_CAT']
  346. all_the_rewards = current_app.config['REWARDS']
  347. if request.method == 'POST':
  348. name = request.form['name']
  349. name = name.lower() # Force the name to all lower case
  350. name = emoji.demojize(name) # Remove any emojis
  351. name = name.replace(" ", "") # Remove any spaces from the name
  352. type = request.form['type']
  353. if name in all_the_rewards: # Check for duplicate reward names
  354. flash("A reward with this name already exists.", "error") # Flash an error message
  355. return redirect(url_for('web_panels.add', reward_type=reward_type)) # Redirect back to the add page
  356. if type != 'category': # If we're only adding a category, skip all of this
  357. cooldown = int(request.form['cooldown'])
  358. if type == 'redeem' or type == 'special' or type == 'vote':
  359. price = int(request.form['price'])
  360. if type == 'goal':
  361. target = int(request.form['target'])
  362. milestone1_desc = request.form['milestone1_desc']
  363. if request.form['milestone1_points'] == '':
  364. milestone1_points = ''
  365. else:
  366. milestone1_points = int(request.form['milestone1_points'])
  367. milestone2_desc = request.form['milestone2_desc']
  368. if request.form['milestone2_points'] == '':
  369. milestone2_points = ''
  370. else:
  371. milestone2_points = int(request.form['milestone2_points'])
  372. milestone3_desc = request.form['milestone3_desc']
  373. if request.form['milestone3_points'] == '':
  374. milestone3_points = ''
  375. else:
  376. milestone3_points = int(request.form['milestone3_points'])
  377. info = request.form['info']
  378. info = emoji.demojize(info) # Remove any emojis
  379. if type == 'special':
  380. cmd = request.form['cmd']
  381. categories = request.form.getlist('category')
  382. if type == 'redeem':
  383. if categories == ['']:
  384. all_the_rewards[name] = {'price': price, 'type': type, 'info': info, 'cooldown': cooldown}
  385. else:
  386. all_the_rewards[name] = {'price': price, 'type': type, 'info': info, 'categories': categories, 'cooldown': cooldown}
  387. if type == 'goal':
  388. if categories == ['']:
  389. all_the_rewards[name] = {'target': target, 'type': type, 'info': info, 'cooldown': cooldown}
  390. else:
  391. all_the_rewards[name] = {'target': target, 'type': type, 'info': info, 'categories': categories, 'cooldown': cooldown}
  392. all_the_rewards[name]["milestones"] = {"milestone1": [], "milestone2": [], "milestone3": []} # Create empty milestones key
  393. if milestone1_points:
  394. all_the_rewards[name]["milestones"]["milestone1"] = [milestone1_desc, milestone1_points]
  395. if milestone2_points:
  396. all_the_rewards[name]["milestones"]["milestone2"] = [milestone2_desc, milestone2_points]
  397. if milestone3_points:
  398. all_the_rewards[name]["milestones"]["milestone3"] = [milestone3_desc, milestone3_points]
  399. if type == 'vote':
  400. if categories == ['']:
  401. all_the_rewards[name] = {'price': price, 'type': type, 'info': info}
  402. else:
  403. all_the_rewards[name] = {'price': price, 'type': type, 'info': info, 'categories': categories, 'cooldown': cooldown}
  404. if type == 'special':
  405. if categories == ['']:
  406. all_the_rewards[name] = {'price': price, 'type': type, 'info': info, 'cmd': cmd, 'cooldown': cooldown}
  407. else:
  408. all_the_rewards[name] = {'price': price, 'type': type, 'info': info, 'cmd': cmd, 'categories': categories, 'cooldown': cooldown}
  409. save_rewards(all_the_rewards)
  410. if type == 'goal': # Remove old goals and votes from the database
  411. reread_goals()
  412. if type == 'vote':
  413. reread_votes()
  414. else: # If we're only adding a category
  415. inactive_categories = current_app.config['INACTIVE_CAT']
  416. inactive_categories.append(name) # Add it to the INACTIVE_CAT variable
  417. reread_categories() # Write it to categories.py
  418. return redirect(url_for('web_panels.mgmt', activeTab='categories'))
  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('/assets/<asset_name>') # Route for displaying alert images
  571. def assets(asset_name):
  572. return send_from_directory(current_app.config['ASSETS_FOLDER'], asset_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)