1
0

web_panels.py 31 KB

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