app.py

import os, json, requests, time, sys
from bottle import route, run, template, static_file, redirect, request, response, put, post, get, error

from search import searchDatabase
from database import Database

dev = False

try:
  database = Database(file='./database/wikiLinks.sqlite')
  print('**** Found deployment database ****')
except:
  database = Database(file='./database/wikiLinksDev.sqlite')
  dev = True

if '-dev' in sys.argv:
  dev = True


if dev:
  print('**** Running in development mode ****')

  @route('/')
  def rootDir():
    redirect('/wikiRace')


@error(404)
def error404(error):
  return template('error404')


@route('/wikiRace/static/<filepath:path>') 
def server_static(filepath):
  return static_file(filepath, root='./assets')


@route('/wikiRace')
def index():
  return template('race')


@route('/wikiRace/search', method='POST')
def search():
  try:
    data = json.loads(request.body.read().decode("utf-8"))
  except Exception as e:
    print(e)

  routes = []
  end = data['end']
  start = data['start']
  initialTime = time.time()
  result = searchDatabase(database, start[1], end[1])
  
  for route in result:
    temp = []
    for pageID in route:
      temp.append(database.getPageName(pageID).replace('\'', ''))  # replaced '[1:]' with '.replace()' as only linux has the extra inverted comma problem
    routes.append(temp)
  
  routes.insert(0, time.time() - initialTime)
  return json.dumps(routes)
  


if __name__ == '__main__':
  port = int(os.environ.get('PORT', 3100))
  ip = '127.0.0.1'
  run(host=ip, port=port, debug=True, reloader=True)


'''
TODO
  some inputs (such as assassins creed) dont fill box with right name and get wrong id
  desanetise inputs
  button to swap start and end?

'''