/
rando-film.py
43 lines (35 loc) · 1.24 KB
/
rando-film.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from flask import Flask, jsonify, request, abort
import os
from os.path import join, dirname
from dotenv import load_dotenv
import urllib
import csv
import random
app = Flask(__name__)
dotenv_path = join(dirname(__file__), '.env')
load_dotenv(dotenv_path)
@app.route('/rando-film/')
def movie_dice():
slack_token = os.environ["SLACK_TOKEN"]
if request.args['token'] and request.args['token'] == slack_token:
# download google spreadsheet
film_url = os.environ['FILM_URL']
urllib.urlretrieve(film_url, 'films.csv')
recs = []
# randomly select a film
with open('films.csv') as film_csv:
reader = csv.reader(film_csv)
next(reader, None) # skip header
for line in reader:
# Collect film titles and recommenders' names
rec = (line[0], line[2])
recs.append(rec)
choice = random.choice(recs)
# send it back in a slack-friendly json format
message = "'%s' recommended by %s" % (choice[0], choice[1])
return jsonify(response_type='in_channel', text=message)
else:
abort(403)
if __name__ == '__main__':
port = int(os.environ.get("PORT", 5000))
app.run(host='0.0.0.0', port=port)