def main():
pp = pprint.PrettyPrinter(indent=2)
creds = get_login_credentials()
power_wall = Powerwall(POWERWALL_HOST)
try:
power_wall.detect_and_pin_version()
except PowerwallUnreachableError as e:
print(e)
return
print("Detected and pinned version: {}".format(
power_wall.get_pinned_version()))
_ = power_wall.login(creds["password"])
print("Current charge: {}".format(power_wall.get_charge()))
print("Device Type: {}".format(power_wall.get_device_type()))
print(f"Authenticated: {power_wall.get_api().is_authenticated()}")
print(f"Sitemaster: {power_wall.get_sitemaster()}")
print(f"Grid status: {power_wall.get_grid_status()}")
print(f"Grid service activer: {power_wall.is_grid_services_active()}")
site_info = power_wall.get_site_info()
print(f"Site info: {pp.pformat(site_info.response)}")
status = power_wall.get_status()
print(f"Status: {pp.pformat(status.response)}")
print(f"Device type: {power_wall.get_device_type()}")
print(f"Serial numbers: {power_wall.get_serial_numbers()}")
print(f"Version: {power_wall.get_version()}")
# # XXX These methods need auth:
print(f"Operation mode: {power_wall.get_operation_mode()}")
# print(f"Backup reserved pct: {power_wall.get_backup_reserved_percentage()}")
print(f"Solars: {power_wall.get_solars()}")
print(f"VIN: {power_wall.get_vin()}")
meters = power_wall.get_meters()
# print(f"Meters: {pp.pformat(meters.response)}")
for meter_type in MeterType:
meter = meters.get_meter(meter_type)
print(f"Meter: {meter_type}")
print(f" Energy exported: {meter.energy_exported}")
print(f" Energy imported: {meter.energy_imported}")
print(f" Instant power: {meter.instant_power}")
print(f" meter info: {pp.pformat(meter)}")
# print(f" Instant reactive power: {meter.instant_reactive_power}")
# print(f" Instant apparent power: {meter.instant_apparent_power}")
# print(f" Instant average voltage: {meter.instant_average_voltage}")
# print(f" Instant total current: {meter.instant_total_current}")
# print(f" Is Active: {meter.is_active()}")
# print(f" Is drawing from: {meter.is_drawing_from()}")
# print(f" Is sending to: {meter.is_sending_to()}")
return
def call_base_info(power_wall: Powerwall, host: str) -> PowerwallBaseInfo:
"""Return PowerwallBaseInfo for the device."""
# Make sure the serial numbers always have the same order
gateway_din = None
with contextlib.suppress(AssertionError, PowerwallError):
gateway_din = power_wall.get_gateway_din().upper()
return PowerwallBaseInfo(
gateway_din=gateway_din,
site_info=power_wall.get_site_info(),
status=power_wall.get_status(),
device_type=power_wall.get_device_type(),
serial_numbers=sorted(power_wall.get_serial_numbers()),
url=f"https://{host}",
)
class TestPowerWall(unittest.TestCase):
def setUp(self):
self.powerwall = Powerwall(ENDPOINT)
def test_get_api(self):
self.assertIsInstance(self.powerwall.get_api(), API)
def test_pins_version_on_creation(self):
pw = Powerwall(ENDPOINT, pin_version="1.49.0")
self.assertEqual(pw.get_pinned_version(),
version.LooseVersion("1.49.0"))
pw = Powerwall(ENDPOINT, pin_version=version.LooseVersion("1.49.0"))
self.assertEqual(pw.get_pinned_version(),
version.LooseVersion("1.49.0"))
@responses.activate
def test_get_charge(self):
add(
Response(GET,
url=f"{ENDPOINT}system_status/soe",
json={"percentage": 53.123423}))
self.assertEqual(self.powerwall.get_charge(), 53.123423)
@responses.activate
def test_get_sitemaster(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}sitemaster",
json=SITEMASTER_RESPONSE))
sitemaster = self.powerwall.get_sitemaster()
self.assertIsInstance(sitemaster, SiteMaster)
self.assertEqual(sitemaster.status, "StatusUp")
self.assertEqual(sitemaster.is_running, True)
self.assertEqual(sitemaster.is_connected_to_tesla, True)
self.assertEqual(sitemaster.is_power_supply_mode, False)
@responses.activate
def test_get_meters(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}meters/aggregates",
json=METERS_AGGREGATES_RESPONSE))
meters = self.powerwall.get_meters()
self.assertIsInstance(meters, MetersAggregates)
self.assertIsInstance(meters.site, Meter)
self.assertIsInstance(meters.solar, Meter)
self.assertIsInstance(meters.battery, Meter)
self.assertIsInstance(meters.load, Meter)
self.assertIsInstance(meters.get_meter(MeterType.SOLAR), Meter)
@responses.activate
def test_meter(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}meters/aggregates",
json=METERS_AGGREGATES_RESPONSE))
@responses.activate
def test_is_sending(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}meters/aggregates",
json=METERS_AGGREGATES_RESPONSE))
meters = self.powerwall.get_meters()
self.assertEqual(meters.solar.is_sending_to(), False)
self.assertEqual(meters.solar.is_active(), True)
self.assertEqual(meters.solar.is_drawing_from(), True)
self.assertEqual(meters.site.is_sending_to(), True)
self.assertEqual(meters.load.is_sending_to(), True)
self.assertEqual(meters.load.is_drawing_from(), False)
self.assertEqual(meters.load.is_active(), True)
@responses.activate
def test_get_grid_status(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}system_status/grid_status",
json=GRID_STATUS_RESPONSE))
grid_status = self.powerwall.get_grid_status()
self.assertEqual(grid_status, GridStatus.CONNECTED)
@responses.activate
def test_is_grid_services_active(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}system_status/grid_status",
json=GRID_STATUS_RESPONSE))
self.assertEqual(self.powerwall.is_grid_services_active(), False)
@responses.activate
def test_get_site_info(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}site_info",
json=SITE_INFO_RESPONSE))
site_info = self.powerwall.get_site_info()
self.assertEqual(site_info.nominal_system_energy, 27)
self.assertEqual(site_info.site_name, "test")
self.assertEqual(site_info.timezone, "Europe/Berlin")
@responses.activate
def test_get_status(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}status",
json=STATUS_RESPONSE))
status = self.powerwall.get_status()
self.assertEqual(
status.up_time_seconds,
datetime.timedelta(seconds=61891, microseconds=214751))
self.assertEqual(
status.start_time,
datetime.datetime(2020,
10,
28,
20,
14,
11,
tzinfo=datetime.timezone(
datetime.timedelta(seconds=28800))))
self.assertEqual(status.device_type, DeviceType.GW1)
self.assertEqual(status.version, "1.50.1")
@responses.activate
def test_get_device_type(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}status",
json=STATUS_RESPONSE))
device_type = self.powerwall.get_device_type()
self.assertIsInstance(device_type, DeviceType)
self.assertEqual(device_type, DeviceType.GW1)
@responses.activate
def test_get_serial_numbers(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}powerwalls",
json=POWERWALLS_RESPONSE))
serial_numbers = self.powerwall.get_serial_numbers()
self.assertEqual(serial_numbers, ["SerialNumber1", "SerialNumber2"])
@responses.activate
def test_get_backup_reserved_percentage(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}operation",
json=OPERATION_RESPONSE))
@responses.activate
def test_get_operation_mode(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}operation",
json=OPERATION_RESPONSE))
@responses.activate
def test_get_version(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}status",
json=STATUS_RESPONSE))
self.assertEqual(self.powerwall.get_version(), "1.50.1")
@responses.activate
def test_detect_and_pin_version(self):
add(
Response(responses.GET,
url=f"{ENDPOINT}status",
json=STATUS_RESPONSE))
vers = version.LooseVersion("1.50.1")
pw = Powerwall(ENDPOINT)
self.assertEqual(pw.detect_and_pin_version(), vers)
self.assertEqual(pw._pin_version, vers)
def test_helpers(self):
resp = {"a": 1}
with self.assertRaises(MissingAttributeError):
assert_attribute(resp, "test")
with self.assertRaises(MissingAttributeError):
assert_attribute(resp, "test", "test")
self.assertEqual(convert_to_kw(2500, -1), 2.5)
write_api = client.write_api(write_options=SYNCHRONOUS)
# Powerwall
ip = os.getenv("POWERWALL_IP")
if ip is None:
raise ValueError("POWERWALL_IP must be set")
email = os.getenv("POWERWALL_EMAIL")
password = os.getenv("POWERWALL_PASSWORD")
power_wall = Powerwall(ip)
# Program
# Identify the powerwall version
power_wall.detect_and_pin_version()
print("Detected and pinned version: {}".format(
power_wall.get_pinned_version()))
print("Current charge: {}".format(power_wall.get_charge()))
print("Device Type: {}".format(power_wall.get_device_type()))
# Sending Data
while True:
p = influxdb_client.Point("Measurement").field("Charge",
power_wall.get_charge)
write_api.write(bucket=bucket, org=org, record=p)
sleep(1)