forked from VenueOne/sportdec-backend-coding-challenge
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathserverTwitter.py
More file actions
105 lines (81 loc) · 2.94 KB
/
serverTwitter.py
File metadata and controls
105 lines (81 loc) · 2.94 KB
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
from BaseHTTPServer import BaseHTTPRequestHandler,HTTPServer
from SocketServer import ThreadingMixIn
import threading
import argparse
import re
import cgi
import tweepy
import json
MAX_TWEETS = 30
class LocalData(object):
records = {}
class Param:
consumer_key = None
consumer_secret = None
def __init__(self):
json_dataKeys=open("keys.json").read()
dataKeys = json.loads(json_dataKeys)
self.consumer_key = dataKeys["consumer_key"]
self.consumer_secret = dataKeys["consumer_secret"]
class Tweepy:
apiTwitter = None
def __init__(self):
param = Param()
auth = tweepy.OAuthHandler(param.consumer_key,param.consumer_secret)
self.apiTwitter = tweepy.API(auth, parser=tweepy.parsers.JSONParser())
def accessTwitterName(self, name):
tweetsJSON = self.apiTwitter.search(q=name, count=MAX_TWEETS)
return tweetsJSON["statuses"] if "statuses" in tweetsJSON.keys() else None
class HTTPRequestHandler(BaseHTTPRequestHandler):
apitweepy = None
def do_GET(self):
if None != re.search('/api/v1/gettweetsname/*', self.path):
name = self.path.split('/')[-1]
allTweets = self.apitweepy.accessTwitterName(name)
if len(allTweets) != None :
self.send_response(200)
self.send_header('Content-Type', 'application/json')
self.send_header('Access-Control-Allow-Origin', '*')
self.end_headers()
self.wfile.write(json.dumps(allTweets))
else:
self.send_response(400, 'Bad Request: record does not exist')
self.send_header('Content-Type', 'application/json')
self.send_header('Access-Control-Allow-Origin', '*')
self.end_headers()
else:
self.send_response(403)
self.send_header('Content-Type', 'application/json')
self.send_header('Access-Control-Allow-Origin', '*')
self.end_headers()
return
class ThreadedHTTPServer(ThreadingMixIn, HTTPServer):
allow_reuse_address = True
def shutdown(self):
self.socket.close()
HTTPServer.shutdown(self)
class SimpleHttpServer():
def __init__(self, ip, port):
apitweepy = Tweepy()
HTTPRequestHandler.apitweepy = apitweepy
self.server = ThreadedHTTPServer((ip,port), HTTPRequestHandler)
def start(self):
self.server_thread = threading.Thread(target=self.server.serve_forever)
self.server_thread.daemon = True
self.server_thread.start()
def waitForThread(self):
self.server_thread.join()
def addRecord(self, recordID, jsonEncodedRecord):
LocalData.records[recordID] = jsonEncodedRecord
def stop(self):
self.server.shutdown()
self.waitForThread()
if __name__=='__main__':
parser = argparse.ArgumentParser(description='HTTP Server')
parser.add_argument('port', type=int, help='Listening port for HTTP Server')
parser.add_argument('ip', help='HTTP Server IP')
args = parser.parse_args()
server = SimpleHttpServer(args.ip, args.port)
print 'HTTP Server Running...........'
server.start()
server.waitForThread()