PulsePoint-Twitter-Bot/bot.py at master · itsJarrett/PulsePoint-Twitter-Bot · GitHub

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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
from datetime import datetime, timedelta
from pytz import timezone
from geolocation.main import GoogleMaps
from credentials import *
import threading
import time
import random
import requests
import tweepy

google_maps = GoogleMaps(api_key='')
auth = tweepy.OAuthHandler(consumer_key, consumer_secret)
auth.set_access_token(access_token, access_token_secret)
api = tweepy.API(auth)
agencyids = [""]
TWEET_SLEEP_TIME = 10
PP_API_SLEEP = 5
MINUTES_REMOVE = 5
currtime = datetime.now()
incidents = []


def grab_pulsepoint(agencyid):
    url = "https://api.pulsepoint.org/v1/incidents"
    pp_json = None
    while pp_json is None:
        try:
            pp_json = requests.get(url,
                                   headers=random_headers(),
                                   params=(
                                       ('apikey', ''),
                                       ('agencyid', agencyid),
                                       ('active', '1'),
                                   ),
                                   timeout=5
                                   ).json()
            return pp_json
        except:
            continue


def get_print_units(units):
    text = ""
    for unit in units:
        text += unit[0] + " [" + unit[1] + "], "
    if text.endswith(", "):
        text = text[:-2]
    return text


def check_if_cleared(agencyid):
    threading.Timer(MINUTES_REMOVE * 60, check_if_cleared, args=[agencyid]).start()
    json = grab_pulsepoint(agencyid)
    if json == "" or json['incidents'] is None: return
    for incident in incidents:
        if not any(incidentid["ID"] in incident[0] for incidentid in json['incidents']):
            incidents.remove(incident)


def status_update_pulsepoint(incidentid, calltype, units):
    for incident in incidents:
        if incident[0] == incidentid:
            if units != incident[2]:
                try:
                    incident[2] = units
                    text = "Units Updated:\n" + get_print_units(units)
                    time.sleep(TWEET_SLEEP_TIME)
                    tweet = api.update_status(text, in_reply_to_status_id=incident[3])
                    print text
                    incident[3] = tweet.id_str
                except Exception as e:
                    print e
            if calltype != incident[1]:
                try:
                    incident[1] = calltype
                    text = "Call Updated to:\n" + calltype
                    time.sleep(TWEET_SLEEP_TIME)
                    tweet = api.update_status(text, in_reply_to_status_id=incident[3])
                    print text
                    incident[3] = tweet.id_str
                except Exception as e:
                    print e


def loop_update_pulsepoint(agencyid):
    threading.Timer(PP_API_SLEEP, loop_update_pulsepoint, args=[agencyid]).start()
    json = grab_pulsepoint(agencyid)
    if json == "" or json['incidents'] is None: return
    for incident in json['incidents']:
        units = []
        if not 'Unit' in incident: return
        for unit in incident['Unit']:
            units.append([unit['UnitID'], next(
                fullstatus for (name, fullstatus) in unit_types if name == unit['PulsePointDispatchStatus'])])
        calltype = next(
            fullname for (name, fullname) in call_types if name == incident['PulsePointIncidentCallType'])
        if not any(incident['ID'] in incident[0] for incident[0] in incidents):
            try:
                address = incident['MedicalEmergencyDisplayAddress']
                dt_obj = datetime.strptime(incident['CallReceivedDateTime'][:-1], "%Y-%m-%dT%H:%M:%S")
                dt = dt_obj.replace(tzinfo=timezone("UTC")).astimezone(timezone("US/Pacific")).strftime(
                    "%m/%d/%y @ %H:%M:%S %Z")
                ht = str(address).split(",")[1].replace(" ", "")
                call = "[" + incident[
                    'IncidentNumber'] + "]" + " CALL: " + calltype + "\nD/T: " + dt + "\nUNITS: " + get_print_units(
                    units) + "\nADDR: " + address + "\n#" + ht
                tweet = api.update_status(call)
                time.sleep(TWEET_SLEEP_TIME)
                incidents.append([incident['ID'], calltype, units, tweet.id_str,
                                  dt_obj.replace(tzinfo=timezone("UTC")).astimezone(timezone("US/Pacific")).replace(
                                      tzinfo=None)])
                print call
            except:
                pass
        else:
            status_update_pulsepoint(incident['ID'], calltype, units)


for agencyid in agencyids:
    loop_update_pulsepoint(agencyid)
    check_if_cleared(agencyid)