alfred-overhere/overify.py at master · monadoid/alfred-overhere · 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
import sys

from pyicloud import PyiCloudService

from workflow import Workflow3

login_info = None
for line in open("apidata.txt", "r").readlines():
    login_info = line.split()

api = PyiCloudService(login_info[0], login_info[1])


def main(wf):
    if api.requires_2sa:
        trusted_devices = api.trusted_devices
        for i, trusted_device in enumerate(trusted_devices):
            # print "  %s: %s" % (i, trusted_device.get('deviceName',
            # "SMS to %s" % trusted_device.get('phoneNumber')))
            wf.add_item(title=trusted_device.get('deviceName'),
                        subtitle='Send iCloud verification code to this device', valid="Yes",
                        arg=trusted_device)


        wf.add_item('Need to verify with iCloud two-factor authentication',
                     "Please use 'overify' keyword to choose which device you'd like to send the code to",
                     valid=False, icon=ICON_WARNING)
        wf.send_feedback()
    else:

        device_list = api.devices._devices

        for device_id, device in device_list.iteritems():
            device_name = str(device).split(":", 1)[1]
            device_hardware = str(device).split(":")[0]
            it = wf.add_item(title=device_name,
                             subtitle=device_hardware, valid="Yes",
                             arg=device_id)
        wf.send_feedback()


if __name__ == "__main__":
    wf = Workflow3()
    sys.exit(wf.run(main))