Black
This commit is contained in:
Paulus Schoutsen
parent
da05dfe708
commit
4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
|
|
@ -11,28 +11,29 @@ import homeassistant.helpers.config_validation as cv
|
|||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
CONF_CLIENT_ID = 'client_id'
|
||||
CONF_CLIENT_SECRET = 'client_secret'
|
||||
CONF_END_LATITUDE = 'end_latitude'
|
||||
CONF_END_LONGITUDE = 'end_longitude'
|
||||
CONF_PRODUCT_IDS = 'product_ids'
|
||||
CONF_START_LATITUDE = 'start_latitude'
|
||||
CONF_START_LONGITUDE = 'start_longitude'
|
||||
CONF_CLIENT_ID = "client_id"
|
||||
CONF_CLIENT_SECRET = "client_secret"
|
||||
CONF_END_LATITUDE = "end_latitude"
|
||||
CONF_END_LONGITUDE = "end_longitude"
|
||||
CONF_PRODUCT_IDS = "product_ids"
|
||||
CONF_START_LATITUDE = "start_latitude"
|
||||
CONF_START_LONGITUDE = "start_longitude"
|
||||
|
||||
ICON = 'mdi:taxi'
|
||||
ICON = "mdi:taxi"
|
||||
|
||||
MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=60)
|
||||
|
||||
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
||||
vol.Required(CONF_CLIENT_ID): cv.string,
|
||||
vol.Required(CONF_CLIENT_SECRET): cv.string,
|
||||
vol.Optional(CONF_START_LATITUDE): cv.latitude,
|
||||
vol.Optional(CONF_START_LONGITUDE): cv.longitude,
|
||||
vol.Optional(CONF_END_LATITUDE): cv.latitude,
|
||||
vol.Optional(CONF_END_LONGITUDE): cv.longitude,
|
||||
vol.Optional(CONF_PRODUCT_IDS):
|
||||
vol.All(cv.ensure_list, [cv.string]),
|
||||
})
|
||||
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
|
||||
{
|
||||
vol.Required(CONF_CLIENT_ID): cv.string,
|
||||
vol.Required(CONF_CLIENT_SECRET): cv.string,
|
||||
vol.Optional(CONF_START_LATITUDE): cv.latitude,
|
||||
vol.Optional(CONF_START_LONGITUDE): cv.longitude,
|
||||
vol.Optional(CONF_END_LATITUDE): cv.latitude,
|
||||
vol.Optional(CONF_END_LONGITUDE): cv.longitude,
|
||||
vol.Optional(CONF_PRODUCT_IDS): vol.All(cv.ensure_list, [cv.string]),
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
def setup_platform(hass, config, add_entities, discovery_info=None):
|
||||
|
|
@ -40,18 +41,22 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
|
|||
from lyft_rides.auth import ClientCredentialGrant
|
||||
from lyft_rides.errors import APIError
|
||||
|
||||
auth_flow = ClientCredentialGrant(client_id=config.get(CONF_CLIENT_ID),
|
||||
client_secret=config.get(
|
||||
CONF_CLIENT_SECRET),
|
||||
scopes="public",
|
||||
is_sandbox_mode=False)
|
||||
auth_flow = ClientCredentialGrant(
|
||||
client_id=config.get(CONF_CLIENT_ID),
|
||||
client_secret=config.get(CONF_CLIENT_SECRET),
|
||||
scopes="public",
|
||||
is_sandbox_mode=False,
|
||||
)
|
||||
try:
|
||||
session = auth_flow.get_session()
|
||||
|
||||
timeandpriceest = LyftEstimate(
|
||||
session, config.get(CONF_START_LATITUDE, hass.config.latitude),
|
||||
session,
|
||||
config.get(CONF_START_LATITUDE, hass.config.latitude),
|
||||
config.get(CONF_START_LONGITUDE, hass.config.longitude),
|
||||
config.get(CONF_END_LATITUDE), config.get(CONF_END_LONGITUDE))
|
||||
config.get(CONF_END_LATITUDE),
|
||||
config.get(CONF_END_LONGITUDE),
|
||||
)
|
||||
timeandpriceest.fetch_data()
|
||||
except APIError as exc:
|
||||
_LOGGER.error("Error setting up Lyft platform: %s", exc)
|
||||
|
|
@ -61,13 +66,11 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
|
|||
|
||||
dev = []
|
||||
for product_id, product in timeandpriceest.products.items():
|
||||
if (wanted_product_ids is not None) and \
|
||||
(product_id not in wanted_product_ids):
|
||||
if (wanted_product_ids is not None) and (product_id not in wanted_product_ids):
|
||||
continue
|
||||
dev.append(LyftSensor('time', timeandpriceest, product_id, product))
|
||||
if product.get('estimate') is not None:
|
||||
dev.append(LyftSensor(
|
||||
'price', timeandpriceest, product_id, product))
|
||||
dev.append(LyftSensor("time", timeandpriceest, product_id, product))
|
||||
if product.get("estimate") is not None:
|
||||
dev.append(LyftSensor("price", timeandpriceest, product_id, product))
|
||||
add_entities(dev, True)
|
||||
|
||||
|
||||
|
|
@ -80,16 +83,15 @@ class LyftSensor(Entity):
|
|||
self._product_id = product_id
|
||||
self._product = product
|
||||
self._sensortype = sensorType
|
||||
self._name = '{} {}'.format(
|
||||
self._product['display_name'], self._sensortype)
|
||||
if 'lyft' not in self._name.lower():
|
||||
self._name = 'Lyft{}'.format(self._name)
|
||||
if self._sensortype == 'time':
|
||||
self._unit_of_measurement = 'min'
|
||||
elif self._sensortype == 'price':
|
||||
estimate = self._product['estimate']
|
||||
self._name = "{} {}".format(self._product["display_name"], self._sensortype)
|
||||
if "lyft" not in self._name.lower():
|
||||
self._name = "Lyft{}".format(self._name)
|
||||
if self._sensortype == "time":
|
||||
self._unit_of_measurement = "min"
|
||||
elif self._sensortype == "price":
|
||||
estimate = self._product["estimate"]
|
||||
if estimate is not None:
|
||||
self._unit_of_measurement = estimate.get('currency')
|
||||
self._unit_of_measurement = estimate.get("currency")
|
||||
self._state = None
|
||||
|
||||
@property
|
||||
|
|
@ -111,40 +113,41 @@ class LyftSensor(Entity):
|
|||
def device_state_attributes(self):
|
||||
"""Return the state attributes."""
|
||||
params = {
|
||||
'Product ID': self._product['ride_type'],
|
||||
'Product display name': self._product['display_name'],
|
||||
'Vehicle Capacity': self._product['seats']
|
||||
"Product ID": self._product["ride_type"],
|
||||
"Product display name": self._product["display_name"],
|
||||
"Vehicle Capacity": self._product["seats"],
|
||||
}
|
||||
|
||||
if self._product.get('pricing_details') is not None:
|
||||
pricing_details = self._product['pricing_details']
|
||||
params['Base price'] = pricing_details.get('base_charge')
|
||||
params['Cancellation fee'] = pricing_details.get(
|
||||
'cancel_penalty_amount')
|
||||
params['Minimum price'] = pricing_details.get('cost_minimum')
|
||||
params['Cost per mile'] = pricing_details.get('cost_per_mile')
|
||||
params['Cost per minute'] = pricing_details.get('cost_per_minute')
|
||||
params['Price currency code'] = pricing_details.get('currency')
|
||||
params['Service fee'] = pricing_details.get('trust_and_service')
|
||||
if self._product.get("pricing_details") is not None:
|
||||
pricing_details = self._product["pricing_details"]
|
||||
params["Base price"] = pricing_details.get("base_charge")
|
||||
params["Cancellation fee"] = pricing_details.get("cancel_penalty_amount")
|
||||
params["Minimum price"] = pricing_details.get("cost_minimum")
|
||||
params["Cost per mile"] = pricing_details.get("cost_per_mile")
|
||||
params["Cost per minute"] = pricing_details.get("cost_per_minute")
|
||||
params["Price currency code"] = pricing_details.get("currency")
|
||||
params["Service fee"] = pricing_details.get("trust_and_service")
|
||||
|
||||
if self._product.get("estimate") is not None:
|
||||
estimate = self._product['estimate']
|
||||
params['Trip distance (in miles)'] = estimate.get(
|
||||
'estimated_distance_miles')
|
||||
params['High price estimate (in cents)'] = estimate.get(
|
||||
'estimated_cost_cents_max')
|
||||
params['Low price estimate (in cents)'] = estimate.get(
|
||||
'estimated_cost_cents_min')
|
||||
params['Trip duration (in seconds)'] = estimate.get(
|
||||
'estimated_duration_seconds')
|
||||
estimate = self._product["estimate"]
|
||||
params["Trip distance (in miles)"] = estimate.get(
|
||||
"estimated_distance_miles"
|
||||
)
|
||||
params["High price estimate (in cents)"] = estimate.get(
|
||||
"estimated_cost_cents_max"
|
||||
)
|
||||
params["Low price estimate (in cents)"] = estimate.get(
|
||||
"estimated_cost_cents_min"
|
||||
)
|
||||
params["Trip duration (in seconds)"] = estimate.get(
|
||||
"estimated_duration_seconds"
|
||||
)
|
||||
|
||||
params['Prime Time percentage'] = estimate.get(
|
||||
'primetime_percentage')
|
||||
params["Prime Time percentage"] = estimate.get("primetime_percentage")
|
||||
|
||||
if self._product.get("eta") is not None:
|
||||
eta = self._product['eta']
|
||||
params['Pickup time estimate (in seconds)'] = eta.get(
|
||||
'eta_seconds')
|
||||
eta = self._product["eta"]
|
||||
params["Pickup time estimate (in seconds)"] = eta.get("eta_seconds")
|
||||
|
||||
return {k: v for k, v in params.items() if v is not None}
|
||||
|
||||
|
|
@ -161,27 +164,39 @@ class LyftSensor(Entity):
|
|||
except KeyError:
|
||||
return
|
||||
self._state = None
|
||||
if self._sensortype == 'time':
|
||||
eta = self._product['eta']
|
||||
if (eta is not None) and (eta.get('is_valid_estimate')):
|
||||
time_estimate = eta.get('eta_seconds')
|
||||
if self._sensortype == "time":
|
||||
eta = self._product["eta"]
|
||||
if (eta is not None) and (eta.get("is_valid_estimate")):
|
||||
time_estimate = eta.get("eta_seconds")
|
||||
if time_estimate is None:
|
||||
return
|
||||
self._state = int(time_estimate / 60)
|
||||
elif self._sensortype == 'price':
|
||||
estimate = self._product['estimate']
|
||||
if (estimate is not None) and \
|
||||
estimate.get('is_valid_estimate'):
|
||||
self._state = (int(
|
||||
(estimate.get('estimated_cost_cents_min', 0) +
|
||||
estimate.get('estimated_cost_cents_max', 0)) / 2) / 100)
|
||||
elif self._sensortype == "price":
|
||||
estimate = self._product["estimate"]
|
||||
if (estimate is not None) and estimate.get("is_valid_estimate"):
|
||||
self._state = (
|
||||
int(
|
||||
(
|
||||
estimate.get("estimated_cost_cents_min", 0)
|
||||
+ estimate.get("estimated_cost_cents_max", 0)
|
||||
)
|
||||
/ 2
|
||||
)
|
||||
/ 100
|
||||
)
|
||||
|
||||
|
||||
class LyftEstimate:
|
||||
"""The class for handling the time and price estimate."""
|
||||
|
||||
def __init__(self, session, start_latitude, start_longitude,
|
||||
end_latitude=None, end_longitude=None):
|
||||
def __init__(
|
||||
self,
|
||||
session,
|
||||
start_latitude,
|
||||
start_longitude,
|
||||
end_latitude=None,
|
||||
end_longitude=None,
|
||||
):
|
||||
"""Initialize the LyftEstimate object."""
|
||||
self._session = session
|
||||
self.start_latitude = start_latitude
|
||||
|
|
@ -194,6 +209,7 @@ class LyftEstimate:
|
|||
def update(self):
|
||||
"""Get the latest product info and estimates from the Lyft API."""
|
||||
from lyft_rides.errors import APIError
|
||||
|
||||
try:
|
||||
self.fetch_data()
|
||||
except APIError as exc:
|
||||
|
|
@ -202,35 +218,41 @@ class LyftEstimate:
|
|||
def fetch_data(self):
|
||||
"""Get the latest product info and estimates from the Lyft API."""
|
||||
from lyft_rides.client import LyftRidesClient
|
||||
|
||||
client = LyftRidesClient(self._session)
|
||||
|
||||
self.products = {}
|
||||
|
||||
products_response = client.get_ride_types(
|
||||
self.start_latitude, self.start_longitude)
|
||||
self.start_latitude, self.start_longitude
|
||||
)
|
||||
|
||||
products = products_response.json.get('ride_types')
|
||||
products = products_response.json.get("ride_types")
|
||||
|
||||
for product in products:
|
||||
self.products[product['ride_type']] = product
|
||||
self.products[product["ride_type"]] = product
|
||||
|
||||
if self.end_latitude is not None and self.end_longitude is not None:
|
||||
price_response = client.get_cost_estimates(
|
||||
self.start_latitude, self.start_longitude,
|
||||
self.end_latitude, self.end_longitude)
|
||||
self.start_latitude,
|
||||
self.start_longitude,
|
||||
self.end_latitude,
|
||||
self.end_longitude,
|
||||
)
|
||||
|
||||
prices = price_response.json.get('cost_estimates', [])
|
||||
prices = price_response.json.get("cost_estimates", [])
|
||||
|
||||
for price in prices:
|
||||
product = self.products[price['ride_type']]
|
||||
product = self.products[price["ride_type"]]
|
||||
if price.get("is_valid_estimate"):
|
||||
product['estimate'] = price
|
||||
product["estimate"] = price
|
||||
|
||||
eta_response = client.get_pickup_time_estimates(
|
||||
self.start_latitude, self.start_longitude)
|
||||
self.start_latitude, self.start_longitude
|
||||
)
|
||||
|
||||
etas = eta_response.json.get('eta_estimates')
|
||||
etas = eta_response.json.get("eta_estimates")
|
||||
|
||||
for eta in etas:
|
||||
if eta.get("is_valid_estimate"):
|
||||
self.products[eta['ride_type']]['eta'] = eta
|
||||
self.products[eta["ride_type"]]["eta"] = eta
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue