def create_app(test_config=None):
# create and configure the app
app = Flask(__name__, instance_relative_config=True)
app.config.from_mapping(SECRET_KEY='dev', )
if test_config is None:
# load the instance config, if it exists, when not testing
app.config.from_pyfile('config.py', silent=True)
else:
# load the test config if passed in
app.config.from_mapping(test_config)
# configure cache buster
cache_buster = CacheBuster(config={
'extensions': ['.js', '.css'],
'hash_size': 5,
})
cache_buster.init_app(app)
# ensure the instance folder exists
try:
os.makedirs(app.instance_path)
except OSError:
pass
# local blueprints
app.register_blueprint(main_bp, url_prefix="/")
for ds in datasets:
app.register_blueprint(dataset.make_blueprint(ds),
url_prefix="/" + ds['id'])
# Google login
blueprint = make_google_blueprint(
client_id=app.config.get('GOOGLE_OAUTH_CLIENT_ID'),
client_secret=app.config.get('GOOGLE_OAUTH_CLIENT_SECRET'),
)
app.register_blueprint(blueprint, url_prefix="/login")
# user handling
app.before_request(add_globals)
# db teardown
app.teardown_appcontext(close_connection)
# cli commands for admins
app.cli.add_command(importer_cli)
# add filter for jinja templating
@app.template_filter('capfirst')
def capfirst(s):
return s[:1].upper() + s[1:]
return app
from flask import Flask, request
from flask import render_template
from flask_cachebuster import CacheBuster
application = Flask(__name__)
#Cache buster
config = {'extensions': ['.js', '.css', '.csv'], 'hash_size': 6}
cache_buster = CacheBuster(config=config)
cache_buster.init_app(application)
appName = 'SemTrack'
@application.route("/")
def index(name="TrackMyCourses"):
return render_template('index.html', name=appName, page="main")
@application.route("/courses/<course_name>")
def course_mark_view(course_name):
return render_template('course.html',
course_name=course_name,
name=appName,
page="calc")
@application.route("/settings")
def settings():
return render_template('settings.html', name=appName, page="settings")
import subprocess
import config
from pprint import pprint
from flask_cors import CORS
if config.RUN_PARSER:
subprocess.Popen(["python", "heroku_parse_runner.py"])
app = Flask(__name__)
CORS(app)
ext = Sitemap(app=app)
cache_buster = CacheBuster(
config={'extensions': ['.js', '.css', '.csv'], 'hash_size': 5})
cache_buster.init_app(app)
# Examples:
universities_example = [
{
"name":'nsu',
"id":0
},
{
"name":'ifmo',
"id":1
},
{
"name":'spbsu',
"id":2
},
def create_app():
app = Flask(__name__,
template_folder='../templates',
static_folder='../static')
app.json_encoder = CustomJSONEncoder
cache = Cache(app,
config={'CACHE_TYPE': 'simple', 'CACHE_DEFAULT_TIMEOUT': 0})
cache_buster = CacheBuster(config={'extensions': ['.js', '.css']})
cache_buster.init_app(app)
Compress(app)
if args.cors:
CORS(app)
db_uri = 'mysql+pymysql://{}:{}@{}:{}/{}?charset=utf8mb4'.format(
args.db_user, args.db_pass, args.db_host, args.db_port, args.db_name)
app.config['SQLALCHEMY_DATABASE_URI'] = db_uri
app.config['SQLALCHEMY_ENGINE_OPTIONS'] = {
'pool_size' : 0 # No limit.
}
app.config['SQLALCHEMY_POOL_RECYCLE'] = args.db_pool_recycle
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db.init_app(app)
if args.client_auth:
app.config['SESSION_TYPE'] = 'redis'
r = redis.Redis(args.redis_host, args.redis_port)
app.config['SESSION_REDIS'] = r
app.config['SESSION_USE_SIGNER'] = True
app.secret_key = args.secret_key
Session(app)
if args.discord_auth:
accepted_auth_types.append('discord')
for config in args.discord_access_configs:
length = len(config.split(':'))
name = (config.split(':')[2] if length == 3 else
config.split(':')[1])
if name not in valid_access_configs:
valid_access_configs.append(name)
if args.telegram_auth:
accepted_auth_types.append('telegram')
for config in args.telegram_access_configs:
name = config.split(':')[1]
if name not in valid_access_configs:
valid_access_configs.append(name)
if not args.disable_blacklist:
log.info('Retrieving blacklist...')
ip_blacklist = get_ip_blacklist()
# Sort & index for binary search
ip_blacklist.sort(key=lambda r: r[0])
ip_blacklist_keys = [
dottedQuadToNum(r[0]) for r in ip_blacklist
]
else:
log.info('Blacklist disabled for this session.')
@app.before_request
def validate_request():
# Get real IP behind trusted reverse proxy.
ip_addr = request.remote_addr
if ip_addr in args.trusted_proxies:
ip_addr = request.headers.get('X-Forwarded-For', ip_addr)
# Make sure IP isn't blacklisted.
if ip_is_blacklisted(ip_addr):
log.debug('Denied access to %s: blacklisted IP.', ip_addr)
abort(403)
if args.client_auth:
session['ip'] = ip_addr
session['last_active'] = datetime.utcnow()
@app.route('/')
@auth_required
def map_page(*_args, **kwargs):
if not kwargs['has_permission']:
return redirect(kwargs['redirect_uri'])
user_args = get_args(kwargs['access_config'])
settings = {
'centerLat': user_args.center_lat,
'centerLng': user_args.center_lng,
'maxZoomLevel': user_args.max_zoom_level,
'showAllZoomLevel': user_args.show_all_zoom_level,
'clusterZoomLevel': user_args.cluster_zoom_level,
'clusterZoomLevelMobile': user_args.cluster_zoom_level_mobile,
'maxClusterRadius': user_args.max_cluster_radius,
'spiderfyClusters': user_args.spiderfy_clusters,
'isStartMarkerMovable': not user_args.lock_start_marker,
'generateImages': user_args.generate_images,
'statsSidebar': not user_args.no_stats_sidebar,
'twelveHourClock': user_args.twelve_hour_clock,
'mapUpdateInverval': user_args.map_update_interval,
'motd': user_args.motd,
'motdTitle': user_args.motd_title,
'motdText': user_args.motd_text,
'motdPages': user_args.motd_pages,
'showMotdAlways': user_args.show_motd_always,
'pokemons': not user_args.no_pokemon,
'upscaledPokemon': (
[int(i) for i in user_args.upscaled_pokemon.split(',')]
if user_args.upscaled_pokemon is not None else []),
'pokemonValues': (not user_args.no_pokemon and
not user_args.no_pokemon_values),
'catchRates': user_args.catch_rates,
'rarity': (not user_args.no_pokemon and user_args.rarity and
user_args.rarity_update_frequency),
'rarityFileName': user_args.rarity_filename,
'pokemonCries': (not user_args.no_pokemon and
user_args.pokemon_cries),
'gyms': not user_args.no_gyms,
'gymSidebar': ((not user_args.no_gyms or
not user_args.no_raids) and
not user_args.no_gym_sidebar),
'gymFilters': (not user_args.no_gyms and
not user_args.no_gym_filters),
'raids': not user_args.no_raids,
'raidFilters': (not user_args.no_raids and
not user_args.no_raid_filters),
'pokestops': not user_args.no_pokestops,
'quests': not user_args.no_pokestops and not user_args.no_quests,
'invasions': (not user_args.no_pokestops and
not user_args.no_invasions),
'lures': not user_args.no_pokestops and not user_args.no_lures,
'weather': not user_args.no_weather,
'spawnpoints': not user_args.no_spawnpoints,
'scannedLocs': not user_args.no_scanned_locs,
's2Cells': not user_args.no_s2_cells,
'ranges': not user_args.no_ranges,
'nestParks': user_args.nest_parks,
'nestParksFileName': user_args.nest_parks_filename,
'exParks': user_args.ex_parks,
'exParksFileName': user_args.ex_parks_filename
}
return render_template(
'map.html',
lang=user_args.locale,
map_title=user_args.map_title,
custom_favicon=user_args.custom_favicon,
header_image=not user_args.no_header_image,
header_image_name=user_args.header_image,
client_auth=user_args.client_auth,
logged_in=is_logged_in(),
admin=is_admin(),
madmin_url=user_args.madmin_url,
donate_url=user_args.donate_url,
patreon_url=user_args.patreon_url,
discord_url=user_args.discord_url,
messenger_url=user_args.messenger_url,
telegram_url=user_args.telegram_url,
whatsapp_url=user_args.whatsapp_url,
pokemon_history_page=(settings['pokemons'] and
not user_args.no_pokemon_history_page),
quest_page=settings['quests'] and not user_args.no_quest_page,
analytics_id=user_args.analytics_id,
settings=settings,
i18n=i8ln
)
@app.route('/pokemon-history')
@auth_required
def pokemon_history_page(*_args, **kwargs):
if not kwargs['has_permission']:
return redirect(kwargs['redirect_uri'])
user_args = get_args(kwargs['access_config'])
if user_args.no_pokemon or user_args.no_pokemon_history_page:
if args.client_auth:
if is_logged_in():
abort(403)
else:
return redirect(url_for('login_page'))
else:
abort(404)
settings = {
'centerLat': user_args.center_lat,
'centerLng': user_args.center_lng,
'generateImages': user_args.generate_images,
'motd': user_args.motd,
'motdTitle': user_args.motd_title,
'motdText': user_args.motd_text,
'motdPages': user_args.motd_pages,
'showMotdAlways': user_args.show_motd_always
}
return render_template(
'pokemon-history.html',
lang=user_args.locale,
map_title=user_args.map_title,
custom_favicon=user_args.custom_favicon,
header_image=not user_args.no_header_image,
header_image_name=user_args.header_image,
client_auth=user_args.client_auth,
logged_in=is_logged_in(),
admin=is_admin(),
madmin_url=user_args.madmin_url,
donate_url=user_args.donate_url,
patreon_url=user_args.patreon_url,
discord_url=user_args.discord_url,
messenger_url=user_args.messenger_url,
telegram_url=user_args.telegram_url,
whatsapp_url=user_args.whatsapp_url,
quest_page=(not user_args.no_pokestops and
not user_args.no_quests and
not user_args.no_quest_page),
analytics_id=user_args.analytics_id,
settings=settings
)
@app.route('/quests')
@auth_required
def quest_page(*_args, **kwargs):
if not kwargs['has_permission']:
return redirect(kwargs['redirect_uri'])
user_args = get_args(kwargs['access_config'])
if (user_args.no_pokestops or user_args.no_quests or
user_args.no_quest_page):
if args.client_auth:
if is_logged_in():
abort(403)
else:
return redirect(url_for('login_page'))
else:
abort(404)
settings = {
'generateImages': user_args.generate_images,
'motd': user_args.motd,
'motdTitle': user_args.motd_title,
'motdText': user_args.motd_text,
'motdPages': user_args.motd_pages,
'showMotdAlways': user_args.show_motd_always
}
return render_template(
'quest.html',
lang=user_args.locale,
map_title=user_args.map_title,
custom_favicon=user_args.custom_favicon,
header_image=not user_args.no_header_image,
header_image_name=user_args.header_image,
client_auth=user_args.client_auth,
logged_in=is_logged_in(),
admin=is_admin(),
madmin_url=user_args.madmin_url,
donate_url=user_args.donate_url,
patreon_url=user_args.patreon_url,
discord_url=user_args.discord_url,
messenger_url=user_args.messenger_url,
telegram_url=user_args.telegram_url,
whatsapp_url=user_args.whatsapp_url,
pokemon_history_page=(not user_args.no_pokemon and
not user_args.no_pokemon_history_page),
analytics_id=user_args.analytics_id,
settings=settings
)
@app.route('/mobile')
@auth_required
def mobile_page(*_args, **kwargs):
if not kwargs['has_permission']:
return redirect(kwargs['redirect_uri'])
user_args = get_args(kwargs['access_config'])
# todo: Check if client is Android/iOS/Desktop for geolink, currently
# only supports Android.
pokemon_list = []
settings = {
'motd': user_args.motd,
'motdTitle': user_args.motd_title,
'motdText': user_args.motd_text,
'motdPages': user_args.motd_pages,
'showMotdAlways': user_args.show_motd_always
}
# Allow client to specify location.
lat = request.args.get('lat', user_args.center_lat, type=float)
lon = request.args.get('lon', user_args.center_lng, type=float)
origin_point = LatLng.from_degrees(lat, lon)
for pokemon in convert_pokemon_list(
Pokemon.get_active(None, None, None, None)):
pokemon_point = LatLng.from_degrees(pokemon['latitude'],
pokemon['longitude'])
diff = pokemon_point - origin_point
diff_lat = diff.lat().degrees
diff_lng = diff.lng().degrees
direction = (('N' if diff_lat >= 0 else 'S')
if abs(diff_lat) > 1e-4 else '') +\
(('E' if diff_lng >= 0 else 'W')
if abs(diff_lng) > 1e-4 else '')
entry = {
'id': pokemon['pokemon_id'],
'name': get_pokemon_name(pokemon['pokemon_id']),
'card_dir': direction,
'distance': int(origin_point.get_distance(
pokemon_point).radians * 6366468.241830914),
'time_to_disappear': '%d min %d sec' % (divmod(
(pokemon['disappear_time'] - datetime.utcnow()).seconds,
60)),
'disappear_time': pokemon['disappear_time'],
'disappear_sec': (
pokemon['disappear_time'] - datetime.utcnow()).seconds,
'latitude': pokemon['latitude'],
'longitude': pokemon['longitude']
}
pokemon_list.append((entry, entry['distance']))
pokemon_list = [y[0] for y in sorted(pokemon_list, key=lambda x: x[1])]
return render_template(
'mobile.html',
custom_favicon=user_args.custom_favicon,
pokemon_list=pokemon_list,
origin_lat=lat,
origin_lng=lon,
analytics_id=user_args.analytics_id,
settings=settings
)
@app.route('/login')
def login_page():
if not args.client_auth:
abort(404)
if is_logged_in():
return redirect(url_for('map_page'))
settings = {
'motd': args.motd,
'motdTitle': args.motd_title,
'motdText': args.motd_text,
'motdPages': args.motd_pages,
'showMotdAlways': args.show_motd_always
}
return render_template(
'login.html',
lang=args.locale,
map_title=args.map_title,
custom_favicon=args.custom_favicon,
header_image=not args.no_header_image,
header_image_name=args.header_image,
madmin_url=args.madmin_url,
donate_url=args.donate_url,
patreon_url=args.patreon_url,
discord_url=args.discord_url,
messenger_url=args.messenger_url,
telegram_url=args.telegram_url,
whatsapp_url=args.whatsapp_url,
analytics_id=args.analytics_id,
discord_auth=args.discord_auth,
telegram_auth=args.telegram_auth,
pokemon_history_page=(not args.no_pokemon and
not args.no_pokemon_history_page),
quest_page=(not args.no_pokestops and not args.no_quests and
not args.no_quest_page),
settings=settings
)
@app.route('/login/<auth_type>')
def login(auth_type):
if not args.client_auth:
abort(404)
if is_logged_in():
return redirect(url_for('map_page'))
if auth_type not in accepted_auth_types:
abort(404)
authenticator = auth_factory.get_authenticator(auth_type)
auth_uri = authenticator.get_authorization_url()
return redirect(auth_uri)
@app.route('/login/telegram')
def telegram_login_page():
if not args.telegram_auth:
abort(404)
settings = {
'motd': args.motd,
'motdTitle': args.motd_title,
'motdText': args.motd_text,
'motdPages': args.motd_pages,
'showMotdAlways': args.show_motd_always
}
return render_template(
'telegram.html',
lang=args.locale,
map_title=args.map_title,
custom_favicon=args.custom_favicon,
header_image=not args.no_header_image,
header_image_name=args.header_image,
madmin_url=args.madmin_url,
donate_url=args.donate_url,
patreon_url=args.patreon_url,
discord_url=args.discord_url,
messenger_url=args.messenger_url,
telegram_url=args.telegram_url,
whatsapp_url=args.whatsapp_url,
pokemon_history_page=(not args.no_pokemon and
not args.no_pokemon_history_page),
quest_page=(not args.no_pokestops and not args.no_quests and
not args.no_quest_page),
analytics_id=args.analytics_id,
telegram_bot_username=args.telegram_bot_username,
server_uri=args.server_uri,
settings=settings
)
@app.route('/auth/<auth_type>')
def auth(auth_type):
if not args.client_auth:
abort(404)
if is_logged_in():
return redirect(url_for('map_page'))
if auth_type not in accepted_auth_types:
abort(404)
auth_factory.get_authenticator(auth_type).authorize()
if args.no_multiple_logins:
r = app.config['SESSION_REDIS']
sessions = get_sessions(r)
for s in sessions:
if 'auth_type' in s and 'id' in s:
if s['auth_type'] == session['auth_type'] and s['id'] == session['id']:
r.delete('session:' + s['session_id'])
return redirect(url_for('map_page'))
@app.route('/logout')
def logout():
if not args.client_auth:
abort(404)
if is_logged_in():
if session['auth_type'] in accepted_auth_types:
a = auth_factory.get_authenticator(session['auth_type'])
a.end_session()
else:
session.clear()
return redirect(url_for('map_page'))
@app.route('/admin')
def admin_page():
return redirect(url_for('users_page'))
@app.route('/admin/users')
@auth_required
def users_page(*_args, **kwargs):
if not args.client_auth:
abort(404)
if not kwargs['has_permission']:
return redirect(kwargs['redirect_uri'])
if not is_admin():
abort(403)
user_args = get_args(kwargs['access_config'])
settings = {
'motd': user_args.motd,
'motdTitle': user_args.motd_title,
'motdText': user_args.motd_text,
'motdPages': user_args.motd_pages,
'showMotdAlways': user_args.show_motd_always
}
return render_template(
'users.html',
lang=user_args.locale,
map_title=user_args.map_title,
custom_favicon=user_args.custom_favicon,
header_image=not user_args.no_header_image,
header_image_name=user_args.header_image,
madmin_url=user_args.madmin_url,
donate_url=user_args.donate_url,
patreon_url=user_args.patreon_url,
discord_url=user_args.discord_url,
messenger_url=user_args.messenger_url,
telegram_url=user_args.telegram_url,
whatsapp_url=user_args.whatsapp_url,
analytics_id=user_args.analytics_id,
pokemon_history_page=(not user_args.no_pokemon and
not user_args.no_pokemon_history_page),
quest_page=(not user_args.no_pokestops and
not user_args.no_quests and
not user_args.no_quest_page),
settings=settings
)
@app.route('/raw-data')
@auth_required
def raw_data(*_args, **kwargs):
if not kwargs['has_permission']:
abort(401)
user_args = get_args(kwargs['access_config'])
# Make sure fingerprint isn't blacklisted.
fingerprint_blacklisted = any([
fingerprints['no_referrer'](request),
fingerprints['iPokeGo'](request)
])
if fingerprint_blacklisted:
log.debug('User denied access: blacklisted fingerprint.')
abort(403)
d = {}
# Request time of this request.
d['timestamp'] = datetime.utcnow()
# Request time of previous request.
if request.args.get('timestamp'):
timestamp = int(request.args.get('timestamp'))
timestamp -= 1000 # Overlap, for rounding errors.
else:
timestamp = 0
swLat = request.args.get('swLat')
swLng = request.args.get('swLng')
neLat = request.args.get('neLat')
neLng = request.args.get('neLng')
oSwLat = request.args.get('oSwLat')
oSwLng = request.args.get('oSwLng')
oNeLat = request.args.get('oNeLat')
oNeLng = request.args.get('oNeLng')
# Previous switch settings.
lastpokemon = request.args.get('lastpokemon')
lastgyms = request.args.get('lastgyms')
lastpokestops = request.args.get('lastpokestops')
lastspawns = request.args.get('lastspawns')
lastscannedlocs = request.args.get('lastscannedlocs')
lastweather = request.args.get('lastweather')
# Current switch settings saved for next request.
if request.args.get('pokemon', 'true') == 'true':
d['lastpokemon'] = True
if (request.args.get('gyms', 'true') == 'true' or
request.args.get('raids', 'true') == 'true'):
d['lastgyms'] = True
if (request.args.get('pokestops', 'true') == 'true' and (
request.args.get('pokestopsNoEvent', 'true') == 'true' or
request.args.get('quests', 'true') == 'true' or
request.args.get('invasions', 'true') == 'true' or
request.args.get('lures', 'true') == 'true')):
d['lastpokestops'] = True
if request.args.get('spawnpoints', 'false') == 'true':
d['lastspawns'] = True
if request.args.get('scannedLocs', 'false') == 'true':
d['lastscannedlocs'] = True
if request.args.get('weather', 'false') == 'true':
d['lastweather'] = True
if (oSwLat is not None and oSwLng is not None and
oNeLat is not None and oNeLng is not None):
# If old coords are not equal to current coords we have
# moved/zoomed!
if (oSwLng < swLng and oSwLat < swLat and
oNeLat > neLat and oNeLng > neLng):
newArea = False # We zoomed in no new area uncovered.
elif not (oSwLat == swLat and oSwLng == swLng and
oNeLat == neLat and oNeLng == neLng):
newArea = True
else:
newArea = False
# Pass current coords as old coords.
d['oSwLat'] = swLat
d['oSwLng'] = swLng
d['oNeLat'] = neLat
d['oNeLng'] = neLng
if (request.args.get('pokemon', 'true') == 'true' and
not user_args.no_pokemon):
verified_despawn = user_args.verified_despawn_time
eids = None
ids = None
if (request.args.get('eids') and
request.args.get('prionotif', 'false') == 'false'):
request_eids = request.args.get('eids').split(',')
eids = [int(i) for i in request_eids]
elif not request.args.get('eids') and request.args.get('ids'):
request_ids = request.args.get('ids').split(',')
ids = [int(i) for i in request_ids]
if lastpokemon != 'true':
# If this is first request since switch on, load
# all pokemon on screen.
d['pokemons'] = convert_pokemon_list(
Pokemon.get_active(
swLat, swLng, neLat, neLng, eids=eids, ids=ids,
verified_despawn_time=verified_despawn))
else:
# If map is already populated only request modified Pokemon
# since last request time.
d['pokemons'] = convert_pokemon_list(
Pokemon.get_active(
swLat, swLng, neLat, neLng, timestamp=timestamp,
eids=eids, ids=ids,
verified_despawn_time=verified_despawn))
if newArea:
# If screen is moved add newly uncovered Pokemon to the
# ones that were modified since last request time.
d['pokemons'] += (
convert_pokemon_list(
Pokemon.get_active(
swLat, swLng, neLat, neLng, oSwLat=oSwLat,
oSwLng=oSwLng, oNeLat=oNeLat, oNeLng=oNeLng,
eids=eids, ids=ids,
verified_despawn_time=verified_despawn)))
if request.args.get('reids'):
request_reids = request.args.get('reids').split(',')
reids = [int(x) for x in request_reids]
d['pokemons'] += convert_pokemon_list(
Pokemon.get_active(swLat, swLng, neLat, neLng, ids=ids,
verified_despawn_time=verified_despawn))
d['reids'] = reids
if request.args.get('seen', 'false') == 'true':
d['seen'] = Pokemon.get_seen(int(request.args.get('duration')))
if request.args.get('appearances', 'false') == 'true':
d['appearances'] = Pokemon.get_appearances(
request.args.get('pokemonid'),
request.args.get('formid'),
int(request.args.get('duration')))
if request.args.get('appearancesDetails', 'false') == 'true':
d['appearancesTimes'] = (
Pokemon.get_appearances_times_by_spawnpoint(
request.args.get('pokemonid'),
request.args.get('spawnpoint_id'),
request.args.get('formid'),
int(request.args.get('duration'))))
gyms = (request.args.get('gyms', 'true') == 'true' and
not user_args.no_gyms)
raids = (request.args.get('raids', 'true') == 'true' and
not user_args.no_raids)
if gyms or raids:
if lastgyms != 'true':
d['gyms'] = Gym.get_gyms(swLat, swLng, neLat, neLng,
raids=raids)
else:
d['gyms'] = Gym.get_gyms(swLat, swLng, neLat, neLng,
timestamp=timestamp, raids=raids)
if newArea:
d['gyms'].update(
Gym.get_gyms(swLat, swLng, neLat, neLng,
oSwLat=oSwLat, oSwLng=oSwLng,
oNeLat=oNeLat, oNeLng=oNeLng,
raids=raids))
pokestops = (request.args.get('pokestops', 'true') == 'true' and
not user_args.no_pokestops)
pokestopsNoEvent = (request.args.get(
'pokestopsNoEvent', 'true') == 'true')
quests = (request.args.get('quests', 'true') == 'true' and
not user_args.no_quests)
invasions = (request.args.get('invasions', 'true') == 'true' and
not user_args.no_invasions)
lures = (request.args.get('lures', 'true') == 'true' and
not user_args.no_lures)
if (pokestops and (pokestopsNoEvent or quests or invasions or lures)):
if lastpokestops != 'true':
d['pokestops'] = Pokestop.get_pokestops(
swLat, swLng, neLat, neLng,
eventless_stops=pokestopsNoEvent, quests=quests,
invasions=invasions, lures=lures
)
else:
d['pokestops'] = Pokestop.get_pokestops(
swLat, swLng, neLat, neLng, timestamp=timestamp,
eventless_stops=pokestopsNoEvent, quests=quests,
invasions=invasions, lures=lures
)
if newArea:
d['pokestops'].update(Pokestop.get_pokestops(
swLat, swLng, neLat, neLng, oSwLat=oSwLat,
oSwLng=oSwLng, oNeLat=oNeLat, oNeLng=oNeLng,
eventless_stops=pokestopsNoEvent, quests=quests,
invasions=invasions, lures=lures
))
if (request.args.get('weather', 'false') == 'true' and
not user_args.no_weather):
if lastweather != 'true':
d['weather'] = Weather.get_weather(swLat, swLng, neLat, neLng)
else:
d['weather'] = Weather.get_weather(swLat, swLng, neLat, neLng,
timestamp=timestamp)
if newArea:
d['weather'] += Weather.get_weather(
swLat, swLng, neLat, neLng, oSwLat=oSwLat,
oSwLng=oSwLng, oNeLat=oNeLat, oNeLng=oNeLng)
if (request.args.get('spawnpoints', 'false') == 'true' and
not user_args.no_spawnpoints):
if lastspawns != 'true':
d['spawnpoints'] = TrsSpawn.get_spawnpoints(
swLat=swLat, swLng=swLng, neLat=neLat, neLng=neLng)
else:
d['spawnpoints'] = TrsSpawn.get_spawnpoints(
swLat=swLat, swLng=swLng, neLat=neLat, neLng=neLng,
timestamp=timestamp)
if newArea:
d['spawnpoints'] += TrsSpawn.get_spawnpoints(
swLat, swLng, neLat, neLng, oSwLat=oSwLat,
oSwLng=oSwLng, oNeLat=oNeLat, oNeLng=oNeLng)
if (request.args.get('scannedLocs', 'false') == 'true' and
not user_args.no_scanned_locs):
if lastscannedlocs != 'true':
d['scannedlocs'] = ScannedLocation.get_recent(
swLat, swLng, neLat, neLng)
else:
d['scannedlocs'] = ScannedLocation.get_recent(
swLat, swLng, neLat, neLng, timestamp=timestamp)
if newArea:
d['scannedlocs'] += ScannedLocation.get_recent(
swLat, swLng, neLat, neLng, oSwLat=oSwLat,
oSwLng=oSwLng, oNeLat=oNeLat, oNeLng=oNeLng)
return jsonify(d)
@app.route('/raw-data/users')
def users_data():
if not args.client_auth:
abort(404)
# Make sure fingerprint isn't blacklisted.
fingerprint_blacklisted = any([
fingerprints['no_referrer'](request),
fingerprints['iPokeGo'](request)
])
if fingerprint_blacklisted:
log.debug('User denied access: blacklisted fingerprint.')
abort(403)
if not is_admin():
abort(403)
sessions = get_sessions(app.config['SESSION_REDIS'])
users = []
for s in sessions:
if 'auth_type' not in s or 'access_data_updated_at' not in s:
continue
del s['_permanent']
del s['has_permission']
del s['access_data_updated_at']
if s['auth_type'] == 'discord':
del s['token']
users.append(s)
return jsonify(users)
@app.route('/pkm_img')
def pokemon_img():
raw = 'raw' in request.args
pkm = int(request.args.get('pkm'))
weather = int(
request.args.get('weather')) if 'weather' in request.args else 0
gender = int(
request.args.get('gender')) if 'gender' in request.args else None
form = int(
request.args.get('form')) if 'form' in request.args else None
costume = int(
request.args.get('costume')) if 'costume' in request.args else None
shiny = 'shiny' in request.args
if raw:
filename = get_pokemon_raw_icon(
pkm, gender=gender, form=form,
costume=costume, weather=weather, shiny=shiny)
else:
filename = get_pokemon_map_icon(
pkm, weather=weather,
gender=gender, form=form, costume=costume)
return send_file(filename, mimetype='image/png')
@app.route('/gym_img')
def gym_img():
team = request.args.get('team')
level = request.args.get('level')
raidlevel = request.args.get('raidlevel')
pkm = request.args.get('pkm')
form = request.args.get('form')
costume = int(
request.args.get('costume')) if 'costume' in request.args else None
is_in_battle = 'in_battle' in request.args
is_ex_raid_eligible = 'is_ex_raid_eligible' in request.args
if level is None or raidlevel is None:
return send_file(
get_gym_icon(team, level, raidlevel, pkm, is_in_battle, form,
costume, is_ex_raid_eligible),
mimetype='image/png'
)
elif (int(level) < 0 or int(level) > 6 or int(raidlevel) < 0 or
int(raidlevel) > 5):
return abort(416)
else:
return send_file(
get_gym_icon(team, level, raidlevel, pkm, is_in_battle, form,
costume, is_ex_raid_eligible),
mimetype='image/png'
)
@app.route('/robots.txt')
def render_robots_txt():
return render_template('robots.txt')
@app.route('/serviceWorker.min.js')
def render_service_worker_js():
return send_from_directory('../static/dist/js', 'serviceWorker.min.js')
return app
def cache_buster():
""" css cache buster """
config = {'extensions': ['.js', '.css'], 'hash_size': 5}
cache_buster = CacheBuster(config=config)
cache_buster.init_app(app)
def main():
# Patch threading to make exceptions catchable.
install_thread_excepthook()
# Make sure exceptions get logged.
sys.excepthook = handle_exception
args = get_args()
# Abort if status name is not valid.
regexp = re.compile('^([\w\s\-.]+)$')
if not regexp.match(args.status_name):
log.critical('Status name contains illegal characters.')
sys.exit(1)
set_log_and_verbosity(log)
args.root_path = os.path.dirname(os.path.abspath(__file__))
init_dynamic_images(args)
# Stop if we're just looking for a debug dump.
if args.dump:
log.info('Retrieving environment info...')
hastebin_id = get_debug_dump_link()
log.info('Done! Your debug link: https://hastebin.com/%s.txt',
hastebin_id)
sys.exit(1)
# Let's not forget to run Grunt.
if not validate_assets(args):
sys.exit(1)
position = extract_coordinates(args.location)
log.info('Parsed location is: %.4f/%.4f/%.4f (lat/lng/alt).', position[0],
position[1], position[2])
# Scanning toggles.
log.info('Parsing of Pokemon %s.',
'disabled' if args.no_pokemon else 'enabled')
log.info('Parsing of Pokestops %s.',
'disabled' if args.no_pokestops else 'enabled')
log.info('Parsing of Gyms %s.', 'disabled' if args.no_gyms else 'enabled')
log.info('Pokemon encounters %s.',
'enabled' if args.encounter else 'disabled')
app = None
if not args.clear_db:
app = Pogom(__name__,
root_path=os.path.dirname(os.path.abspath(__file__)))
app.before_request(app.validate_request)
app.set_location(position)
db = startup_db(app, args.clear_db)
# Database cleaner; really only need one ever.
if args.db_cleanup:
t = Thread(target=clean_db_loop, name='db-cleaner', args=(args, ))
t.daemon = True
t.start()
# Dynamic rarity.
if args.rarity_update_frequency:
t = Thread(target=dynamic_rarity_refresher, name='dynamic-rarity')
t.daemon = True
log.info('Dynamic rarity is enabled.')
t.start()
else:
log.info('Dynamic rarity is disabled.')
# Parks downloading
if args.parks:
t = Thread(target=download_all_parks, name='parks')
t.daemon = True
log.info('Parks downloading is enabled.')
t.start()
else:
log.info('Parks downloading is disabled.')
if args.cors:
CORS(app)
# No more stale JS.
cache_buster = CacheBuster()
cache_buster.init_app(app)
ssl_context = None
if (args.ssl_certificate and args.ssl_privatekey
and os.path.exists(args.ssl_certificate)
and os.path.exists(args.ssl_privatekey)):
ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLSv1_2)
ssl_context.load_cert_chain(args.ssl_certificate, args.ssl_privatekey)
log.info('Web server in SSL mode.')
if args.verbose:
app.run(threaded=True,
use_reloader=False,
debug=True,
host=args.host,
port=args.port,
ssl_context=ssl_context)
else:
app.run(threaded=True,
use_reloader=False,
debug=False,
host=args.host,
port=args.port,
ssl_context=ssl_context)
def cache_buster():
""" static url cache buster, to load new css and js """
config = { 'extensions': ['.js', '.css'], 'hash_size': 5 }
cache_buster = CacheBuster(config=config)
cache_buster.init_app(app)