forked from tdorssers/TeslaPy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
cli.py
executable file
·172 lines (166 loc) · 8.47 KB
/
cli.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
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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
""" Tesla Owner API CLI application using TeslaPy module """
# Author: Tim Dorssers
from __future__ import print_function
import ast
import logging
import argparse
try:
import webview # Optional pywebview 3.0 or higher
except ImportError:
webview = None
try:
from selenium import webdriver # Optional selenium 3.13.0 or higher
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait
except ImportError:
webdriver = None
from teslapy import Tesla, Vehicle, Battery, SolarPanel
raw_input = vars(__builtins__).get('raw_input', input) # Py2/3 compatibility
def custom_auth(url):
# Use pywebview if no web browser specified
if webview and not (webdriver and args.web is not None):
result = ['']
window = webview.create_window('Login', url)
def on_loaded():
result[0] = window.get_current_url()
if 'void/callback' in result[0].split('?')[0]:
window.destroy()
try:
window.events.loaded += on_loaded
except AttributeError:
window.loaded += on_loaded
webview.start()
return result[0]
# Use selenium to control specified web browser
options = [webdriver.chrome, webdriver.opera,
webdriver.edge][args.web].options.Options()
options.add_argument('--disable-blink-features=AutomationControlled')
with [webdriver.Chrome, webdriver.Opera,
webdriver.Edge][args.web](options=options) as browser:
logging.info('Selenium opened %s', browser.capabilities['browserName'])
browser.get(url)
WebDriverWait(browser, 300).until(EC.url_contains('void/callback'))
return browser.current_url
def main():
default_format = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
logging.basicConfig(level=logging.DEBUG if args.debug else logging.INFO,
format=default_format)
with Tesla(args.email, verify=args.verify, proxy=args.proxy,
sso_base_url=args.url) as tesla:
if (webdriver and args.web is not None) or webview:
tesla.authenticator = custom_auth
if args.timeout:
tesla.timeout = args.timeout
if args.refresh:
tesla.refresh_token(refresh_token=args.refresh)
selected = prod = tesla.vehicle_list() + tesla.battery_list() + tesla.solar_list()
if args.filter:
selected = [p for p in prod for v in p.values() if v == args.filter]
logging.info('%d product(s), %d selected', len(prod), len(selected))
for i, product in enumerate(selected):
print('Product %d:' % i)
# Show information or invoke API depending on arguments
if args.list:
print(product)
if isinstance(product, Vehicle):
if args.vin:
print(product.decode_vin())
if args.wake:
product.sync_wake_up()
if args.get:
print(product.get_vehicle_data())
if args.nearby:
print(product.get_nearby_charging_sites())
if args.mobile:
print(product.mobile_enabled())
if args.stream:
product.stream(print)
if args.service:
print(product.get_service_scheduling_data())
if args.history:
print(product.get_charge_history())
elif isinstance(product, Battery) and args.battery:
print(product.get_battery_data())
elif isinstance(product, SolarPanel) and args.site:
print(product.get_site_data())
if args.api or args.command:
data = {}
for key, value in args.keyvalue or []:
try:
data[key] = ast.literal_eval(value)
except (SyntaxError, ValueError):
data[key] = value
if args.api:
for commandargs in args.api:
command = commandargs.pop(0)
command_data = {}
for keyvalue in commandargs or []:
key, value = keyvalue.split('=', 1)
try:
command_data[key] = ast.literal_eval(value)
except (SyntaxError, ValueError):
command_data[key] = value
if not command_data:
command_data = data
print(product.api(command, **command_data))
else:
print(product.command(args.command, **data))
if args.logout:
if webview and not (webdriver and args.web is not None):
window = webview.create_window('Logout', tesla.logout())
webview.start()
else:
tesla.logout(not (webdriver and args.web is not None))
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Tesla Owner API CLI')
parser.add_argument('-e', dest='email', help='login email', required=True)
parser.add_argument('-f', dest='filter', help='filter on id, vin, etc.')
parser.add_argument('-a', dest='api', help='API call endpoint name',
metavar=('API', 'KEYVALUE'), action='append', nargs='+')
parser.add_argument('-k', dest='keyvalue', help='API parameter (key=value)',
action='append', type=lambda kv: kv.split('=', 1))
parser.add_argument('-c', dest='command', help='product command endpoint')
parser.add_argument('-t', dest='timeout', type=int,
help='connect/read timeout')
parser.add_argument('-p', dest='proxy', help='proxy server URL')
parser.add_argument('-R', dest='refresh', help='use this refresh token')
parser.add_argument('-U', dest='url', help='SSO service base URL')
parser.add_argument('-l', '--list', action='store_true',
help='list all selected vehicles/batteries')
parser.add_argument('-v', '--vin', action='store_true',
help='vehicle identification number decode')
parser.add_argument('-w', '--wake', action='store_true',
help='wake up selected vehicle(s)')
parser.add_argument('-g', '--get', action='store_true',
help='get rollup of all vehicle data')
parser.add_argument('-b', '--battery', action='store_true',
help='get detailed battery state and config')
parser.add_argument('-n', '--nearby', action='store_true',
help='list nearby charging sites')
parser.add_argument('-m', '--mobile', action='store_true',
help='get mobile enabled state')
parser.add_argument('-s', '--site', action='store_true',
help='get current site generation data')
parser.add_argument('-d', '--debug', action='store_true',
help='set logging level to debug')
parser.add_argument('-r', '--stream', action='store_true',
help='receive streaming vehicle data on-change')
parser.add_argument('-S', '--service', action='store_true',
help='get service self scheduling eligibility')
parser.add_argument('-H', '--history', action='store_true',
help='get charging history data')
parser.add_argument('-V', '--verify', action='store_false',
help='disable verify SSL certificate')
parser.add_argument('-L', '--logout', action='store_true',
help='clear token from cache and logout')
if webdriver:
h = 'use Chrome browser' if webview else 'use Chrome browser (default)'
parser.add_argument('--chrome', action='store_const', dest='web',
help=h, const=0, default=None if webview else 0)
parser.add_argument('--opera', action='store_const', dest='web',
help='use Opera browser', const=1)
if hasattr(webdriver.edge, 'options'):
parser.add_argument('--edge', action='store_const', dest='web',
help='use Edge browser', const=2)
args = parser.parse_args()
main()