app.py 34 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058
  1. import datetime as datetime
  2. from urllib.parse import urlparse
  3. import webbrowser
  4. from emailtool.emailer import Emailer
  5. from flask import Flask, render_template, request, redirect, url_for, session, send_file
  6. from gevent.pywsgi import WSGIServer
  7. import socket
  8. import logging
  9. from peewee import Model, CharField, DateTimeField, SqliteDatabase, BooleanField, fn
  10. import hashlib
  11. import os
  12. import csv
  13. import time
  14. import threading
  15. import atexit
  16. db_name = 'app.db'
  17. db = SqliteDatabase(db_name)
  18. class User(Model):
  19. username = CharField()
  20. password = CharField()
  21. date_created = DateTimeField(default=datetime.datetime.now)
  22. logged_in = BooleanField(default=False)
  23. class Meta:
  24. database = db
  25. class IlsUser(Model):
  26. username = CharField()
  27. email = CharField()
  28. reset_datetime = DateTimeField(default=datetime.datetime.now)
  29. class Meta:
  30. database = db
  31. class Settings(Model):
  32. name = CharField()
  33. value = CharField()
  34. class Meta:
  35. database = db
  36. class Schedule(Model):
  37. interval = CharField()
  38. class Meta:
  39. database = db
  40. class Log(Model):
  41. date = DateTimeField(default=datetime.datetime.now)
  42. username = CharField()
  43. action = CharField()
  44. class Meta:
  45. database = db
  46. class PasswordResetLog(Model):
  47. date = DateTimeField(default=datetime.datetime.now)
  48. username = CharField()
  49. class Meta:
  50. database = db
  51. class EmailReminderLog(Model):
  52. date = CharField(default=datetime.datetime.now().strftime('%Y-%m-%d'))
  53. username = CharField()
  54. email = CharField()
  55. interval = CharField()
  56. sent = BooleanField(default=False)
  57. ignore = BooleanField(default=False)
  58. class Meta:
  59. database = db
  60. def scheduler(stop_event):
  61. while not stop_event.is_set():
  62. db.connect()
  63. # Get all scheduled intervals from the DB
  64. intervals = Schedule.select().execute()
  65. users = IlsUser.select().execute()
  66. # Get settings
  67. domain_name_setting = Settings.get(Settings.name == 'Domain Name')
  68. domain_name = domain_name_setting.value
  69. password_reset_interval_setting = Settings.get(Settings.name == 'Password Reset Interval')
  70. password_reset_interval = password_reset_interval_setting.value
  71. smtp_host_setting = Settings.get(Settings.name == 'SMTP Host')
  72. smtp_host = smtp_host_setting.value
  73. smtp_port_setting = Settings.get(Settings.name == 'SMTP Port')
  74. smtp_port = smtp_port_setting.value
  75. smtp_username_setting = Settings.get(Settings.name == 'SMTP Username')
  76. smtp_username = smtp_username_setting.value
  77. smtp_password_setting = Settings.get(Settings.name == 'SMTP Password')
  78. smtp_password = smtp_password_setting.value
  79. # Loop through each user and calculate the number of days until the users password
  80. # expires based on the password_reset_interval value.
  81. # Loop through all users
  82. # Schedule emails to be sent for each user and interval
  83. for user in users:
  84. # get the scheduled intervals
  85. for interval in intervals:
  86. # Calculate the date (yyyy-mm-dd) that the password will expire and subtract the interval
  87. # from that date to get the date that the email should be sent.
  88. # Example: If the password reset interval is 90 days and the current date is 2019-01-01
  89. # then the password will expire on 2019-04-01. If the interval is 30 days then the
  90. # email should be sent on 2019-03-02.
  91. # Calculate the date that the password will expire
  92. password_expiration_date = user.reset_datetime + datetime.timedelta(days=int(password_reset_interval))
  93. # Subtract the interval from the password expiration date to get the date that the email should be sent
  94. email_date = password_expiration_date - datetime.timedelta(days=int(interval.interval))
  95. # check if reminder has already been saved to the DB
  96. reminder = EmailReminderLog.select().where(EmailReminderLog.date == email_date.strftime('%Y-%m-%d'), EmailReminderLog.username == user.username, EmailReminderLog.interval == interval.interval).execute()
  97. if reminder:
  98. continue
  99. else:
  100. # Store the date in the DB
  101. EmailReminderLog.create(date=email_date.strftime('%Y-%m-%d'), email=user.email, username=user.username, interval=interval.interval).save()
  102. # Get all email reminders from the DB
  103. # if today's date matches the EmailReminderLog date then send the email and mark the reminder as sent
  104. reminders = EmailReminderLog.select().where(EmailReminderLog.date == datetime.datetime.now().strftime('%Y-%m-%d'), EmailReminderLog.ignore == False).execute()
  105. for reminder in reminders:
  106. # Send the email
  107. emailer = Emailer(smtp_host, smtp_port, smtp_username, smtp_password)
  108. emailer.send_email(reminder.email, 'Password Reset Reminder', 'Your password will expire in ' + reminder.interval + ' days. Please visit ' + domain_name + ' to reset your password.')
  109. # Mark the reminder as sent
  110. reminder.sent = True
  111. reminder.save()
  112. db.close()
  113. time.sleep(10) # Sleep for 15 minutes
  114. # Encrypt the password with SHA256
  115. def encrypt_password(password):
  116. hash = hashlib.sha256(password.encode('utf-8')).hexdigest()
  117. return hash
  118. # Check for DB tables and create if they don't exist
  119. if db.table_exists('user') is False:
  120. db.create_tables([User, ])
  121. User.create(username='admin', password=encrypt_password('admin'),
  122. date_created=datetime.datetime.now(), logged_in=False).save()
  123. if db.table_exists('ilsuser') is False:
  124. db.create_tables([IlsUser, ])
  125. if db.table_exists('settings') is False:
  126. db.create_tables([Settings, ])
  127. Settings.create(name='Debug Mode', value=False).save()
  128. Settings.create(name='HTTP Port', value=5055).save()
  129. Settings.create(name='SMTP Host', value="").save()
  130. Settings.create(name='SMTP Port', value="587").save()
  131. Settings.create(name='SMTP Username', value="").save()
  132. Settings.create(name='SMTP Password', value="").save()
  133. Settings.create(name='Domain Name', value="lynx").save()
  134. Settings.create(name='Password Reset URL', value="https://terminal.idaho-lynx.org").save()
  135. Settings.create(name='Password Reset Interval', value="90").save()
  136. if db.table_exists('log') is False:
  137. db.create_tables([Log, ])
  138. if db.table_exists('password_reset_log') is False:
  139. db.create_tables([PasswordResetLog, ])
  140. if db.table_exists('schedule') is False:
  141. db.create_tables([Schedule, ])
  142. if db.table_exists('email_reminder_log') is False:
  143. db.create_tables([EmailReminderLog, ])
  144. db.close()
  145. settings = Settings.select().execute()
  146. debug_setting = Settings.get(Settings.name == 'Debug Mode')
  147. debug = debug_setting.value
  148. http_port_setting = Settings.get(Settings.name == 'HTTP Port')
  149. http_port = http_port_setting.value
  150. domain_name_setting = Settings.get(Settings.name == 'Domain Name')
  151. domain_name = domain_name_setting.value
  152. password_reset_url_setting = Settings.get(Settings.name == 'Password Reset URL')
  153. password_reset_url = password_reset_url_setting.value
  154. password_reset_interval_setting = Settings.get(Settings.name == 'Password Reset Interval')
  155. password_reset_interval = password_reset_interval_setting.value
  156. log = logging.getLogger('werkzeug')
  157. log.setLevel(logging.INFO)
  158. if debug.lower() == 'true':
  159. debug = True
  160. else:
  161. debug = False
  162. # send_email('aday@twinfallspubliclibrary.org', 'TEST', 'This is a test email')
  163. def shutdown_session(exception=None):
  164. print('Stopping HTTP Service...')
  165. http_server.stop()
  166. # Get the systems hostname
  167. def get_hostname():
  168. return socket.gethostname()
  169. # Get systems IP address
  170. def get_ip_address():
  171. return socket.gethostbyname(socket.gethostname())
  172. # Method to check if a URL is valid using regex
  173. def is_valid_url(url):
  174. try:
  175. result = urlparse(url)
  176. if all([result.scheme, result.netloc]):
  177. url = '{uri.scheme}://{uri.netloc}/'.format(uri=result)
  178. else:
  179. url = False
  180. return url
  181. except:
  182. return False
  183. def requires_auth():
  184. if 'username' in session:
  185. username = session['username']
  186. user = User.get(User.username == username)
  187. if user.logged_in is True:
  188. return True
  189. else:
  190. return False
  191. else:
  192. return False
  193. def admin_password_check():
  194. admin_user = User.get(User.username == 'admin')
  195. if admin_user.password == encrypt_password('admin'):
  196. return True
  197. db.close()
  198. # Start the scheduler loop in another thread
  199. shutdown_scheduler = threading.Event()
  200. scheduler_thread = threading.Thread(target=scheduler, args=(shutdown_scheduler,))
  201. scheduler_thread.start()
  202. # Start the HTTP Server
  203. app = Flask(__name__)
  204. app.secret_key = os.urandom(24)
  205. def format_time_ago(timestamp):
  206. """Calculate the time passed since a datetime stamp and format it as a human-readable string."""
  207. now = datetime.datetime.now()
  208. diff = now - timestamp
  209. if diff.days > 365:
  210. years = diff.days // 365
  211. return f"{years} year{'s' if years > 1 else ''} ago"
  212. if diff.days > 30:
  213. months = diff.days // 30
  214. return f"{months} month{'s' if months > 1 else ''} ago"
  215. if diff.days > 0:
  216. return f"{diff.days} day{'s' if diff.days > 1 else ''} ago"
  217. if diff.seconds > 3600:
  218. hours = diff.seconds // 3600
  219. return f"{hours} hour{'s' if hours > 1 else ''} ago"
  220. if diff.seconds > 60:
  221. minutes = diff.seconds // 60
  222. return f"{minutes} minute{'s' if minutes > 1 else ''} ago"
  223. return "just now"
  224. # Create method for time until
  225. def format_time_until(timestamp):
  226. """Calculate the time until a datetime stamp and format it as a human-readable string."""
  227. # Convert timestap into datetime object
  228. timestamp = datetime.datetime.strptime(timestamp, '%Y-%m-%d')
  229. now = datetime.datetime.now()
  230. diff = timestamp - now
  231. if diff.days > 365:
  232. years = diff.days // 365
  233. return f"in {years} year{'s' if years > 1 else ''}"
  234. # return months and days
  235. if diff.days > 30:
  236. months = diff.days // 30
  237. days = diff.days % 30
  238. return f"in {months} month{'s' if months > 1 else ''} and {days} day{'s' if days > 1 else ''}"
  239. # return days
  240. if diff.days > 0:
  241. return f"in {diff.days} day{'s' if diff.days > 1 else ''}"
  242. # return hours and minutes
  243. if diff.seconds > 3600:
  244. hours = diff.seconds // 3600
  245. minutes = (diff.seconds % 3600) // 60
  246. return f"in {hours} hour{'s' if hours > 1 else ''} and {minutes} minute{'s' if minutes > 1 else ''}"
  247. # return minutes
  248. if diff.seconds > 60:
  249. minutes = diff.seconds // 60
  250. return f"in {minutes} minute{'s' if minutes > 1 else ''}"
  251. return "now"
  252. if diff.days > 30:
  253. months = diff.days // 30
  254. return f"in {months} month{'s' if months > 1 else ''}"
  255. if diff.days > 0:
  256. return f"in {diff.days} day{'s' if diff.days > 1 else ''}"
  257. if diff.seconds > 3600:
  258. hours = diff.seconds // 3600
  259. return f"in {hours} hour{'s' if hours > 1 else ''}"
  260. if diff.seconds > 60:
  261. minutes = diff.seconds // 60
  262. return f"in {minutes} minute{'s' if minutes > 1 else ''}"
  263. return "now"
  264. app.jinja_env.filters['time_since'] = format_time_ago
  265. app.jinja_env.filters['time_until'] = format_time_until
  266. @app.before_request
  267. def before_request():
  268. db.connect()
  269. @app.after_request
  270. def after_request(response):
  271. db.close()
  272. return response
  273. @app.route('/admin/password/reset', methods=['GET', 'POST'])
  274. def admin_password_reset():
  275. message = None
  276. if request.method == 'POST':
  277. password = request.form.get('password')
  278. password_confirm = request.form.get('password_confirm')
  279. if password != password_confirm:
  280. message = 'Passwords do not match'
  281. else:
  282. try:
  283. user = User.get(User.username == 'admin')
  284. user.password = encrypt_password(password)
  285. user.logged_in = True
  286. user.save()
  287. session['username'] = 'admin'
  288. return redirect(url_for('admin'))
  289. except Exception as e:
  290. print(e)
  291. message = 'Username not found'
  292. context = {
  293. 'message': message,
  294. }
  295. return render_template('admin_password_reset.html', context=context)
  296. # Create a route for the home page
  297. @app.route('/', methods=['GET', 'POST'])
  298. def index():
  299. error = None
  300. reset = False
  301. reset_url = is_valid_url(password_reset_url)
  302. reset_url_error = False
  303. if reset_url is False:
  304. reset_url_error = True
  305. if request.method == 'POST':
  306. username = request.form.get('username')
  307. # Check for the username in the DB
  308. try:
  309. user = IlsUser.filter(IlsUser.username == username).first()
  310. except Exception as e:
  311. print(e)
  312. user = None
  313. if user:
  314. # Reset login datetime
  315. user.reset_datetime = datetime.datetime.now()
  316. PasswordResetLog.create(username=user.username, date_created=datetime.datetime.now()).save()
  317. user.save()
  318. # Open the reset URL in a new tab if the URL is valid
  319. if reset_url is not False:
  320. webbrowser.open_new_tab(str(reset_url))
  321. # Set reset to True to pass back to the view to display the correct content back to the user.
  322. reset = True
  323. else:
  324. error = 'Invalid username'
  325. context = {
  326. 'domain': domain_name,
  327. 'error': error,
  328. 'reset': reset,
  329. 'reset_url': reset_url,
  330. 'reset_url_error': reset_url_error,
  331. }
  332. return render_template('index.html', context=context)
  333. # Create a route for admin page
  334. @app.route('/admin/')
  335. def admin():
  336. if admin_password_check():
  337. return redirect(url_for('admin_password_reset'))
  338. # Check to see if user is logged in
  339. if not requires_auth():
  340. return redirect(url_for('login'))
  341. # Get a count of all IlsUsers
  342. try:
  343. ils_users = IlsUser.select().count()
  344. except Exception as e:
  345. ils_users = None
  346. # Get a count of all EmailReminders that are not ignored or sent
  347. try:
  348. total_email_reminders = EmailReminderLog.select().where(EmailReminderLog.sent == False, EmailReminderLog.ignore == False).count()
  349. except Exception as e:
  350. total_email_reminders = None
  351. # Get a list of coming email reminders for the next 7 days
  352. try:
  353. email_reminders = PasswordResetLog.select().where(PasswordResetLog.date_created < datetime.datetime.now() + datetime.timedelta(days=7)).execute()
  354. except Exception as e:
  355. email_reminders = None
  356. # Get a list ils users that have passwords expiring in the next 7 days
  357. try:
  358. ils_users_expiring = IlsUser.select().where(IlsUser.password_expires < datetime.datetime.now() + datetime.timedelta(days=7)).execute()
  359. except Exception as e:
  360. ils_users_expiring = None
  361. # get a list of email notifications sent in the last 7 days
  362. try:
  363. # Get a list of all future email reminders
  364. email_reminders = EmailReminderLog.select().order_by(-EmailReminderLog.date).limit(15).execute()
  365. except Exception as e:
  366. email_reminders = None
  367. context = {
  368. 'ils_user_count': ils_users,
  369. 'ils_users_expiring': ils_users_expiring,
  370. 'email_reminders': email_reminders,
  371. 'total_email_reminders': total_email_reminders,
  372. }
  373. return render_template('admin.html', context=context)
  374. @app.route('/admin/users/', methods=['GET', 'POST'])
  375. def admin_users():
  376. # Check to see if user is logged in
  377. if not requires_auth():
  378. return redirect(url_for('login'))
  379. message = None
  380. if request.method == 'POST':
  381. username = request.form.get('username')
  382. password = request.form.get('password')
  383. confirm_password = request.form.get('confirm_password')
  384. # Check to see if username already exists
  385. try:
  386. user = User.filter(User.username == username).first()
  387. except Exception as e:
  388. print(e)
  389. user = None
  390. if user:
  391. message = 'Username already exists'
  392. else:
  393. if password == confirm_password:
  394. User.create(username=username, password=encrypt_password(password),
  395. date_created=datetime.datetime.now(), logged_in=False).save()
  396. message = 'User created successfully'
  397. Log.create(username=session['username'], action='Created admin user: %s' % username, ).save()
  398. else:
  399. message = 'Passwords do not match'
  400. # Get all admin users from the DB
  401. users = User.select().execute()
  402. context = {
  403. 'users': users,
  404. 'message': message,
  405. }
  406. return render_template('admin_users.html', context=context)
  407. @app.route('/admin/users/edit/<int:id>', methods=['GET', 'POST'])
  408. def admin_users_edit(id):
  409. # Check to see if user is logged in
  410. if not requires_auth():
  411. return redirect(url_for('login'))
  412. # Get the user from the DB
  413. user = User.get(User.id == id)
  414. message = None
  415. if request.method == 'POST':
  416. username = request.form.get('username')
  417. password = request.form.get('password')
  418. confirm_password = request.form.get('confirm_password')
  419. # Check to see if username already exists
  420. all_users = list()
  421. users = User.select().execute()
  422. for u in users:
  423. if u.username != user.username:
  424. all_users.append(u.username)
  425. if username in all_users:
  426. message = 'Username already exists'
  427. else:
  428. user.username = username
  429. if password is not None or password != '':
  430. if password == confirm_password:
  431. user.password = encrypt_password(password)
  432. else:
  433. message = 'Passwords do not match'
  434. user.save()
  435. message = 'User updated successfully'
  436. Log.create(username=session['username'], action='Updated admin user: %s' % username, ).save()
  437. context = {
  438. 'user': user,
  439. 'message': message,
  440. }
  441. return render_template('admin_user_edit.html', context=context)
  442. @app.route('/admin/users/delete/<int:id>', methods=['GET', 'POST'])
  443. def admin_users_delete(id):
  444. # Check to see if user is logged in
  445. if not requires_auth():
  446. return redirect(url_for('login'))
  447. # Get the user from the DB
  448. user = User.get(User.id == id)
  449. if user.username != 'admin':
  450. # Unset session variable if the user is deleting their own account
  451. if user.username == session['username']:
  452. session.pop('username', None)
  453. username = user.username
  454. user.delete_instance()
  455. Log.create(username=session['username'], action='Removed admin user: %s' % username, ).save()
  456. return redirect(url_for('admin_users'))
  457. @app.route('/admin/users/ils', methods=['GET', 'POST'])
  458. def admin_ils_users():
  459. # Check to see if user is logged in
  460. if not requires_auth():
  461. return redirect(url_for('login'))
  462. message = None
  463. if request.method == 'POST':
  464. username = request.form.get('username')
  465. email = request.form.get('email')
  466. # Check to see if username already exists
  467. try:
  468. user = IlsUser.filter(IlsUser.username == username).first()
  469. except Exception as e:
  470. print(e)
  471. user = None
  472. if user:
  473. message = 'Username already exists'
  474. else:
  475. IlsUser.create(username=username, email=email, reset_datetime=datetime.datetime.now()).save()
  476. message = 'ILS User: %s created successfully' % username
  477. Log.create(username=session['username'], action='Created ILS User: %s' % username, ).save()
  478. # Get all admin users from the DB
  479. users = IlsUser.select().execute()
  480. context = {
  481. 'users': users,
  482. 'message': message,
  483. }
  484. return render_template('admin_ils_users.html', context=context)
  485. @app.route('/admin/users/ils/delete/<int:id>', methods=['GET', 'POST'])
  486. def admin_ils_users_delete(id):
  487. # Check to see if user is logged in
  488. if not requires_auth():
  489. return redirect(url_for('login'))
  490. # Get the user from the DB
  491. user = IlsUser.get(IlsUser.id == id)
  492. username = user.username
  493. # Remove scheduled email reminders
  494. try:
  495. email_reminders = EmailReminderLog.select().where(EmailReminderLog.username == user.username).execute()
  496. for reminder in email_reminders:
  497. reminder.delete_instance()
  498. print('here')
  499. except Exception as e:
  500. print('there', e)
  501. user.delete_instance()
  502. Log.create(username=session['username'], action='Removed ILS user: %s' % username, ).save()
  503. return redirect(url_for('admin_ils_users'))
  504. @app.route('/admin/users/ils/edit/<int:id>', methods=['GET', 'POST'])
  505. def admin_ils_users_edit(id):
  506. # Check to see if user is logged in
  507. if not requires_auth():
  508. return redirect(url_for('login'))
  509. # Get the user from the DB
  510. user = IlsUser.get(IlsUser.id == id)
  511. message = None
  512. if request.method == 'POST':
  513. username = request.form.get('username')
  514. email = request.form.get('email')
  515. # Check to see if username already exists
  516. all_users = list()
  517. users = IlsUser.select().execute()
  518. for u in users:
  519. if u.username != user.username:
  520. all_users.append(u.username)
  521. if username in all_users:
  522. message = 'Username already exists'
  523. else:
  524. # Update Scheduled Email reminders
  525. try:
  526. email_reminders = EmailReminderLog.select().where(EmailReminderLog.username == user.username).execute()
  527. for reminder in email_reminders:
  528. reminder.username = username
  529. reminder.email = email
  530. reminder.save()
  531. except Exception as e:
  532. print(e)
  533. user.username = username
  534. user.email = email
  535. user.save()
  536. message = 'User updated successfully'
  537. Log.create(username=session['username'], action='Updated ILS user: %s' % username, ).save()
  538. context = {
  539. 'user': user,
  540. 'message': message,
  541. }
  542. return render_template('admin_ils_user_edit.html', context=context)
  543. # create a route for generating a CSV file for download
  544. @app.route('/admin/users/ils/csv/download', methods=['GET', 'POST'])
  545. def admin_ils_users_csv_download():
  546. # Check to see if user is logged in
  547. if not requires_auth():
  548. return redirect(url_for('login'))
  549. # Create a CSV file with the users and don't add a blank line between rows
  550. with open('users.csv', 'w', newline='') as f:
  551. writer = csv.writer(f)
  552. writer.writerow(['username', 'email'])
  553. users = IlsUser.select().execute()
  554. for user in users:
  555. writer.writerow([user.username, user.email])
  556. Log.create(username=session['username'], action='Downloaded ILS user CSV file.').save()
  557. # return the CSV file to the user
  558. return send_file('users.csv', as_attachment=True)
  559. @app.route('/admin/users/ils/csv/import', methods=['GET', 'POST'])
  560. def admin_ils_users_csv_import():
  561. # Check to see if user is logged in
  562. if not requires_auth():
  563. return redirect(url_for('login'))
  564. message = None
  565. if request.method == 'POST':
  566. csv_file = request.files['csv']
  567. if csv_file.filename != '':
  568. csv_file.save(os.path.join('uploads', csv_file.filename))
  569. with open(os.path.join('uploads', csv_file.filename), 'r') as f:
  570. reader = csv.reader(f)
  571. for row in reader:
  572. username = row[0]
  573. email = row[1]
  574. # ignore the header row
  575. if username == 'username':
  576. continue
  577. # ignore blank rows
  578. if username == '':
  579. continue
  580. # Check if user already exists and if it does update the entry
  581. try:
  582. user = IlsUser.filter(IlsUser.username == username).first()
  583. except Exception as e:
  584. print(e)
  585. user = None
  586. if user:
  587. user.email = email
  588. user.reset_datetime = datetime.datetime.now()
  589. user.save()
  590. else:
  591. IlsUser.create(username=username, email=email, reset_datetime=datetime.datetime.now()).save()
  592. # Delete the uploaded file
  593. os.remove(os.path.join('uploads', csv_file.filename))
  594. return redirect(url_for('admin_ils_users'))
  595. context = {
  596. 'message': message,
  597. }
  598. return render_template('csv.html', context=context)
  599. @app.route('/admin/settings', methods=['GET', 'POST'])
  600. def settings():
  601. # Check to see if user is logged in
  602. if not requires_auth():
  603. return redirect(url_for('login'))
  604. message = None
  605. # Process form submission
  606. if request.method == 'POST':
  607. # Assign form values to variables
  608. id = request.form.get('id')
  609. value = request.form.get('value')
  610. # Check if the setting exists
  611. try:
  612. setting = Settings.get(Settings.id == id)
  613. except Exception as e:
  614. print(e)
  615. setting = None
  616. if setting:
  617. # Scrub the values based on the setting name
  618. if setting.name == 'Debug Mode':
  619. if value.lower() == 'true':
  620. value = True
  621. else:
  622. value = False
  623. value = bool(value)
  624. elif setting.name == 'HTTP Port':
  625. value = str(value)
  626. elif setting.name == 'SMTP Host':
  627. value = str(value)
  628. elif setting.name == 'SMTP Port':
  629. value = str(value)
  630. elif setting.name == 'SMTP Username':
  631. value = str(value)
  632. elif setting.name == 'SMTP Password':
  633. value = str(value)
  634. elif setting.name == 'Domain Name':
  635. value = str(value)
  636. elif setting.name == 'Password Reset URL':
  637. value = str(value)
  638. elif setting.name == 'Password Reset Interval':
  639. if value.isdigit():
  640. value = int(value)
  641. else:
  642. value = setting.value
  643. # Update the setting
  644. old_value = setting.value
  645. setting.value = value
  646. setting.save()
  647. Log.create(username=session['username'], action='Changed %s setting from "%s" to "%s"' % (setting.name,
  648. old_value,
  649. value)).save()
  650. message = '%s updated successfully' % setting.name
  651. # Get settings from DB
  652. all_settings = Settings.select().execute()
  653. context = {
  654. 'settings': all_settings,
  655. 'message': message,
  656. }
  657. return render_template('settings.html', context=context)
  658. @app.route('/admin/schedule', methods=['GET', 'POST'])
  659. def schedule():
  660. # Check to see if user is logged in
  661. if not requires_auth():
  662. return redirect(url_for('login'))
  663. message = None
  664. # Get all schedules from the DB
  665. schedules = Schedule.select().order_by(Schedule.interval.cast("INTEGER")).execute()
  666. # add schedule
  667. if request.method == 'POST':
  668. # Assign form values to variables
  669. interval = request.form.get('interval')
  670. # Check if interval is a number
  671. try:
  672. int(interval)
  673. except Exception as e:
  674. print(e)
  675. message = 'Error creating schedule. Value submitted must be a whole number.'
  676. return render_template('schedule.html', context={'message': message, 'schedules': schedules})
  677. # Check if the schedule already exists
  678. try:
  679. schedule = Schedule.get(Schedule.interval == interval)
  680. except Exception as e:
  681. print(e)
  682. schedule = None
  683. if schedule:
  684. message = 'Schedule already exists'
  685. else:
  686. # Create the schedule
  687. Schedule.create(interval=interval).save()
  688. Log.create(username=session['username'], action='Created schedule for %s day interval.' % interval).save()
  689. message = 'Schedule: %s created successfully' % interval
  690. schedules = Schedule.select().order_by(Schedule.interval.cast("INTEGER")).execute()
  691. context = {
  692. 'schedules': schedules,
  693. 'message': message,
  694. }
  695. return render_template('schedule.html', context=context)
  696. @app.route('/admin/schedule/emails')
  697. def scheduled_emails():
  698. # Check to see if user is logged in
  699. if not requires_auth():
  700. return redirect(url_for('login'))
  701. # Get all future emailreminderlogs
  702. reminders = EmailReminderLog.select().where(EmailReminderLog.date > datetime.datetime.now(), EmailReminderLog.ignore == False).order_by(
  703. -EmailReminderLog.date).execute()
  704. context = {
  705. 'reminders': reminders,
  706. }
  707. return render_template('scheduled_emails.html', context=context)
  708. # remove schedule
  709. @app.route('/admin/schedule/remove/reminder/<int:id>')
  710. def reminder_remove(id):
  711. # Check to see if user is logged in
  712. if not requires_auth():
  713. return redirect(url_for('login'))
  714. # Get EmailReminderLogs with the interval of the schedule being removed
  715. email_reminder_log = EmailReminderLog.get(EmailReminderLog.id == id)
  716. Log.create(username=session['username'],
  717. action='Removed %s day reminder for %s (%s)' % (email_reminder_log.interval, email_reminder_log.username, email_reminder_log.email)).save()
  718. email_reminder_log.ignore = True
  719. email_reminder_log.save()
  720. return redirect(url_for('scheduled_emails'))
  721. # remove schedule
  722. @app.route('/admin/schedule/remove/<int:id>', methods=['GET', 'POST'])
  723. def schedule_remove(id):
  724. # Check to see if user is logged in
  725. if not requires_auth():
  726. return redirect(url_for('login'))
  727. # Get Schedule by id
  728. try:
  729. schedule = Schedule.get(Schedule.id == id)
  730. except Exception as e:
  731. schedule = None
  732. if schedule:
  733. # Get EmailReminderLogs with the interval of the schedule being removed
  734. email_reminder_logs = EmailReminderLog.select().where(EmailReminderLog.interval == schedule.interval).execute()
  735. for email_reminder_log in email_reminder_logs:
  736. email_reminder_log.delete_instance()
  737. schedule.delete_instance()
  738. Log.create(username=session['username'], action='Removed schedule for a %s day reminder.' % schedule.interval).save()
  739. return redirect(url_for('schedule'))
  740. @app.route('/admin/system/log')
  741. def system_log():
  742. # Check to see if user is logged in
  743. if not requires_auth():
  744. return redirect(url_for('login'))
  745. # Get all logs from the DB
  746. logs = Log.select().order_by(Log.id.desc()).execute()
  747. context = {
  748. 'logs': logs,
  749. }
  750. return render_template('system_log.html', context=context)
  751. @app.route('/admin/system/log/password/resets')
  752. def password_reset_log():
  753. # Check to see if user is logged in
  754. if not requires_auth():
  755. return redirect(url_for('login'))
  756. # Get all logs from the DB
  757. logs = PasswordResetLog.select().order_by(PasswordResetLog.id.desc()).execute()
  758. context = {
  759. 'logs': logs,
  760. }
  761. return render_template('password_reset_log.html', context=context)
  762. @app.route('/logout')
  763. def logout():
  764. if 'username' in session:
  765. username = session['username']
  766. user = User.get(User.username == username)
  767. user.logged_in = False
  768. Log.create(username=session['username'], action='Logged out').save()
  769. user.save()
  770. session.pop('username', None)
  771. return redirect(url_for('login'))
  772. @app.route('/login', methods=['GET', 'POST'])
  773. def login():
  774. if request.method == 'POST':
  775. username = request.form.get('username')
  776. password = encrypt_password(request.form.get('password'))
  777. try:
  778. user = User.filter(User.username == username and User.password == password).first()
  779. except Exception as e:
  780. print(e)
  781. user = None
  782. session.pop('username', None)
  783. if user:
  784. # Login user
  785. session['username'] = request.form.get('username')
  786. user.logged_in = True
  787. Log.create(username=session['username'], action='Logged in').save()
  788. user.save()
  789. return redirect(url_for('admin'))
  790. else:
  791. error = 'Invalid Credentials. Please try again.'
  792. context = {
  793. 'error': error
  794. }
  795. return render_template('login.html', context=context)
  796. context = {
  797. }
  798. return render_template('login.html', context=context)
  799. def clean_up():
  800. shutdown_scheduler.set()
  801. http_server.stop()
  802. if __name__ == "__main__":
  803. print("------------------------- Start up -----------------------------")
  804. print("Starting HTTP Service on port %s..." % http_port)
  805. if debug is True:
  806. print("Debug mode is enabled.")
  807. http_server = WSGIServer(('0.0.0.0', int(http_port)), app)
  808. else:
  809. http_server = WSGIServer(('0.0.0.0', int(http_port)), app, log=log, error_log=log)
  810. print("HTTP Service Started.")
  811. print("--------------------- Application Details ---------------------")
  812. print("Application started at %s" % datetime.datetime.now())
  813. print("System IP Address: %s" % get_ip_address())
  814. print("System Hostname: %s" % get_hostname())
  815. print("Access the Dashboard using a web browser using any of the following:")
  816. print("http://%s:%s or http://%s:%s" % (get_hostname(), http_port, get_ip_address(), http_port))
  817. print("---------------------------------------------------------------")
  818. print("To stop the application close this window.")
  819. print("---------------------------------------------------------------")
  820. http_server.serve_forever()
  821. atexit.register(clean_up)