123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811 |
- # SPDX-License-Identifier: Apache-2.0
- # Copyright 2020 Contributors to OpenLEADR
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- # http://www.apache.org/licenses/LICENSE-2.0
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- """
- OpenADR Client for Python
- """
- import asyncio
- import inspect
- import logging
- import ssl
- import random
- from datetime import datetime, timedelta, timezone
- from functools import partial
- from http import HTTPStatus
- import aiohttp
- from lxml.etree import XMLSyntaxError
- from signxml.exceptions import InvalidSignature
- from apscheduler.schedulers.asyncio import AsyncIOScheduler
- from openleadr import enums, objects, errors
- from openleadr.messaging import create_message, parse_message, \
- validate_xml_schema, validate_xml_signature
- from openleadr import utils
- logger = logging.getLogger('openleadr')
- class OpenADRClient:
- """
- Main client class. Most of these methods will be called automatically, but
- you can always choose to call them manually.
- """
- def __init__(self, ven_name, vtn_url, debug=False, cert=None, key=None,
- passphrase=None, vtn_fingerprint=None, show_fingerprint=True, ca_file=None,
- allow_jitter=True):
- """
- Initializes a new OpenADR Client (Virtual End Node)
- :param str ven_name: The name for this VEN
- :param str vtn_url: The URL of the VTN (Server) to connect to
- :param bool debug: Whether or not to print debugging messages
- :param str cert: The path to a PEM-formatted Certificate file to use
- for signing messages.
- :param str key: The path to a PEM-formatted Private Key file to use
- for signing messages.
- :param str fingerprint: The fingerprint for the VTN's certificate to
- verify incomnig messages
- :param str show_fingerprint: Whether to print your own fingerprint
- on startup. Defaults to True.
- :param str ca_file: The path to the PEM-formatted CA file for validating the VTN server's
- certificate.
- """
- self.ven_name = ven_name
- if vtn_url.endswith("/"):
- vtn_url = vtn_url[:-1]
- self.vtn_url = vtn_url
- self.ven_id = None
- self.registration_id = None
- self.poll_frequency = None
- self.vtn_fingerprint = vtn_fingerprint
- self.debug = debug
- self.reports = []
- self.report_callbacks = {} # Holds the callbacks for each specific report
- self.report_requests = [] # Keep track of the report requests from the VTN
- self.incomplete_reports = {} # Holds reports that are being populated over time
- self.pending_reports = asyncio.Queue() # Holds reports that are waiting to be sent
- self.scheduler = AsyncIOScheduler()
- self.client_session = None
- self.report_queue_task = None
- self.responded_events = {} # Holds the events that we already saw.
- self.cert_path = cert
- self.key_path = key
- self.passphrase = passphrase
- self.ca_file = ca_file
- self.allow_jitter = allow_jitter
- if cert and key:
- with open(cert, 'rb') as file:
- cert = file.read()
- with open(key, 'rb') as file:
- key = file.read()
- if show_fingerprint:
- print("")
- print("*" * 80)
- print("Your VEN Certificate Fingerprint is ".center(80))
- print(f"{utils.certificate_fingerprint(cert).center(80)}".center(80))
- print("Please deliver this fingerprint to the VTN.".center(80))
- print("You do not need to keep this a secret.".center(80))
- print("*" * 80)
- print("")
- self._create_message = partial(create_message,
- cert=cert,
- key=key,
- passphrase=passphrase)
- async def run(self):
- """
- Run the client in full-auto mode.
- """
- # if not hasattr(self, 'on_event'):
- # raise NotImplementedError("You must implement on_event.")
- self.loop = asyncio.get_event_loop()
- await self.create_party_registration()
- if not self.ven_id:
- logger.error("No VEN ID received from the VTN, aborting.")
- await self.stop()
- return
- if self.reports:
- await self.register_reports(self.reports)
- self.report_queue_task = self.loop.create_task(self._report_queue_worker())
- await self._poll()
- # Set up automatic polling
- if self.poll_frequency > timedelta(hours=24):
- logger.warning("Polling with intervals of more than 24 hours is not supported. "
- "Will use 24 hours as the logging interval.")
- self.poll_frequency = timedelta(hours=24)
- cron_config = utils.cron_config(self.poll_frequency, randomize_seconds=self.allow_jitter)
- self.scheduler.add_job(self._poll,
- trigger='cron',
- **cron_config)
- self.scheduler.start()
- async def stop(self):
- """
- Cleanly stops the client. Run this coroutine before closing your event loop.
- """
- if self.scheduler.running:
- self.scheduler.shutdown()
- if self.report_queue_task:
- self.report_queue_task.cancel()
- await self.client_session.close()
- def add_handler(self, handler, callback):
- """
- Add a callback for the given situation
- """
- if handler not in ('on_event', 'on_update_event'):
- logger.error("'handler' must be either on_event or on_update_event")
- return
- setattr(self, handler, callback)
- def add_report(self, callback, resource_id, measurement=None,
- data_collection_mode='incremental',
- report_specifier_id=None, r_id=None,
- report_name=enums.REPORT_NAME.TELEMETRY_USAGE,
- reading_type=enums.READING_TYPE.DIRECT_READ,
- report_type=enums.REPORT_TYPE.READING, sampling_rate=None, data_source=None,
- scale="none", unit=None, power_ac=True, power_hertz=50, power_voltage=230,
- market_context=None):
- """
- Add a new reporting capability to the client.
- :param callable callback: A callback or coroutine that will fetch the value for a specific
- report. This callback will be passed the report_id and the r_id
- of the requested value.
- :param str resource_id: A specific name for this resource within this report.
- :param str measurement: The quantity that is being measured (openleadr.enums.MEASUREMENTS).
- Optional for TELEMETRY_STATUS reports.
- :param str data_collection_mode: Whether you want the data to be collected incrementally
- or at once. If the VTN requests the sampling interval to be
- higher than the reporting interval, this setting determines
- if the callback should be called at the sampling rate (with
- no args, assuming it returns the current value), or at the
- reporting interval (with date_from and date_to as keyword
- arguments). Choose 'incremental' for the former case, or
- 'full' for the latter case.
- :param str report_specifier_id: A unique identifier for this report. Leave this blank for a
- random generated id, or fill it in if your VTN depends on
- this being a known value, or if it needs to be constant
- between restarts of the client.
- :param str r_id: A unique identifier for a datapoint in a report. The same remarks apply as
- for the report_specifier_id.
- :param str report_name: An OpenADR name for this report (one of openleadr.enums.REPORT_NAME)
- :param str reading_type: An OpenADR reading type (found in openleadr.enums.READING_TYPE)
- :param str report_type: An OpenADR report type (found in openleadr.enums.REPORT_TYPE)
- :param datetime.timedelta sampling_rate: The sampling rate for the measurement.
- :param str unit: The unit for this measurement.
- """
- # Verify input
- if report_name not in enums.REPORT_NAME.values and not report_name.startswith('x-'):
- raise ValueError(f"{report_name} is not a valid report_name. Valid options are "
- f"{', '.join(enums.REPORT_NAME.values)}",
- " or any name starting with 'x-'.")
- if reading_type not in enums.READING_TYPE.values and not reading_type.startswith('x-'):
- raise ValueError(f"{reading_type} is not a valid reading_type. Valid options are "
- f"{', '.join(enums.READING_TYPE.values)}"
- " or any name starting with 'x-'.")
- if report_type not in enums.REPORT_TYPE.values and not report_type.startswith('x-'):
- raise ValueError(f"{report_type} is not a valid report_type. Valid options are "
- f"{', '.join(enums.REPORT_TYPE.values)}"
- " or any name starting with 'x-'.")
- if scale not in enums.SI_SCALE_CODE.values:
- raise ValueError(f"{scale} is not a valid scale. Valid options are "
- f"{', '.join(enums.SI_SCALE_CODE.values)}")
- if sampling_rate is None:
- sampling_rate = objects.SamplingRate(min_period=timedelta(seconds=10),
- max_period=timedelta(hours=24),
- on_change=False)
- elif isinstance(sampling_rate, timedelta):
- sampling_rate = objects.SamplingRate(min_period=sampling_rate,
- max_period=sampling_rate,
- on_change=False)
- if data_collection_mode not in ('incremental', 'full'):
- raise ValueError("The data_collection_mode should be 'incremental' or 'full'.")
- if data_collection_mode == 'full':
- args = inspect.signature(callback).parameters
- if not ('date_from' in args and 'date_to' in args and 'sampling_interval' in args):
- raise TypeError("Your callback function must accept the 'date_from', 'date_to' "
- "and 'sampling_interval' arguments if used "
- "with data_collection_mode 'full'.")
- # Determine the correct item name, item description and unit
- if report_name == 'TELEMETRY_STATUS':
- item_base = None
- elif isinstance(measurement, objects.Measurement):
- item_base = measurement
- elif isinstance(measurement, dict):
- utils.validate_report_measurement_dict(measurement)
- power_attributes = object.PowerAttributes(**measurement.get('power_attributes')) or None
- item_base = objects.Measurement(name=measurement['name'],
- description=measurement['description'],
- unit=measurement['unit'],
- scale=measurement.get('scale'),
- power_attributes=power_attributes)
- elif measurement.upper() in enums.MEASUREMENTS.members:
- item_base = enums.MEASUREMENTS[measurement.upper()]
- else:
- item_base = objects.Measurement(name='customUnit',
- description=measurement,
- unit=unit,
- scale=scale)
- if report_name != 'TELEMETRY_STATUS' and scale is not None:
- if item_base.scale is not None:
- if scale in enums.SI_SCALE_CODE.values:
- item_base.scale = scale
- else:
- raise ValueError("The 'scale' argument must be one of '{'. ',join(enums.SI_SCALE_CODE.values)}")
- # Check if unit is compatible
- if unit is not None and unit != item_base.unit and unit not in item_base.acceptable_units:
- logger.warning(f"The supplied unit {unit} for measurement {measurement} "
- f"will be ignored, {item_base.unit} will be used instead. "
- f"Allowed units for this measurement are: "
- f"{', '.join(item_base.acceptable_units)}")
- # Get or create the relevant Report
- if report_specifier_id:
- report = utils.find_by(self.reports,
- 'report_name', report_name,
- 'report_specifier_id', report_specifier_id)
- else:
- report = utils.find_by(self.reports, 'report_name', report_name)
- if not report:
- report_specifier_id = report_specifier_id or utils.generate_id()
- report = objects.Report(created_date_time=datetime.now(),
- report_name=report_name,
- report_specifier_id=report_specifier_id,
- data_collection_mode=data_collection_mode)
- self.reports.append(report)
- # Add the new report description to the report
- target = objects.Target(resource_id=resource_id)
- r_id = utils.generate_id()
- report_description = objects.ReportDescription(r_id=r_id,
- reading_type=reading_type,
- report_data_source=target,
- report_subject=target,
- report_type=report_type,
- sampling_rate=sampling_rate,
- measurement=item_base,
- market_context='Market01')
- self.report_callbacks[(report.report_specifier_id, r_id)] = callback
- report.report_descriptions.append(report_description)
- ###########################################################################
- # #
- # POLLING METHODS #
- # #
- ###########################################################################
- async def poll(self):
- """
- Request the next available message from the Server. This coroutine is called automatically.
- """
- service = 'OadrPoll'
- message = self._create_message('oadrPoll', ven_id=self.ven_id)
- response_type, response_payload = await self._perform_request(service, message)
- return response_type, response_payload
- ###########################################################################
- # #
- # REGISTRATION METHODS #
- # #
- ###########################################################################
- async def query_registration(self):
- """
- Request information about the VTN.
- """
- request_id = utils.generate_id()
- service = 'EiRegisterParty'
- message = self._create_message('oadrQueryRegistration', request_id=request_id)
- response_type, response_payload = await self._perform_request(service, message)
- return response_type, response_payload
- async def create_party_registration(self, http_pull_model=True, xml_signature=False,
- report_only=False, profile_name='2.0b',
- transport_name='simpleHttp', transport_address=None,
- ven_id=None):
- """
- Take the neccessary steps to register this client with the server.
- :param bool http_pull_model: Whether to use the 'pull' model for HTTP.
- :param bool xml_signature: Whether to sign each XML message.
- :param bool report_only: Whether or not this is a reporting-only client
- which does not deal with Events.
- :param str profile_name: Which OpenADR profile to use.
- :param str transport_name: The transport name to use. Either 'simpleHttp' or 'xmpp'.
- :param str transport_address: Which public-facing address the server should use
- to communicate.
- :param str ven_id: The ID for this VEN. If you leave this blank,
- a VEN_ID will be assigned by the VTN.
- """
- request_id = utils.generate_id()
- service = 'EiRegisterParty'
- payload = {'ven_name': self.ven_name,
- 'http_pull_model': http_pull_model,
- 'xml_signature': xml_signature,
- 'report_only': report_only,
- 'profile_name': profile_name,
- 'transport_name': transport_name,
- 'transport_address': transport_address}
- if ven_id:
- payload['ven_id'] = ven_id
- message = self._create_message('oadrCreatePartyRegistration',
- request_id=request_id,
- **payload)
- response_type, response_payload = await self._perform_request(service, message)
- if response_type is None:
- return
- if response_payload['response']['response_code'] != 200:
- status_code = response_payload['response']['response_code']
- status_description = response_payload['response']['response_description']
- logger.error(f"Got error on Create Party Registration: "
- f"{status_code} {status_description}")
- return
- self.ven_id = response_payload['ven_id']
- self.registration_id = response_payload['registration_id']
- self.poll_frequency = response_payload.get('requested_oadr_poll_freq',
- timedelta(seconds=10))
- logger.info(f"VEN is now registered with ID {self.ven_id}")
- logger.info(f"The polling frequency is {self.poll_frequency}")
- return response_type, response_payload
- async def cancel_party_registration(self):
- raise NotImplementedError("Cancel Registration is not yet implemented")
- ###########################################################################
- # #
- # EVENT METHODS #
- # #
- ###########################################################################
- async def request_event(self, reply_limit=1):
- """
- Request the next Event from the VTN, if it has any.
- """
- payload = {'request_id': utils.generate_id(),
- 'ven_id': self.ven_id,
- 'reply_limit': reply_limit}
- message = self._create_message('oadrRequestEvent', **payload)
- service = 'EiEvent'
- response_type, response_payload = await self._perform_request(service, message)
- return response_type, response_payload
- async def created_event(self, request_id, event_id, opt_type, modification_number=1):
- """
- Inform the VTN that we created an event.
- """
- service = 'EiEvent'
- payload = {'ven_id': self.ven_id,
- 'response': {'response_code': 200,
- 'response_description': 'OK',
- 'request_id': request_id},
- 'event_responses': [{'response_code': 200,
- 'response_description': 'OK',
- 'request_id': request_id,
- 'event_id': event_id,
- 'modification_number': modification_number,
- 'opt_type': opt_type}]}
- message = self._create_message('oadrCreatedEvent', **payload)
- response_type, response_payload = await self._perform_request(service, message)
- ###########################################################################
- # #
- # REPORTING METHODS #
- # #
- ###########################################################################
- async def register_reports(self, reports):
- """
- Tell the VTN about our reports. The VTN miht respond with an
- oadrCreateReport message that tells us which reports are to be sent.
- """
- request_id = utils.generate_id()
- payload = {'request_id': request_id,
- 'ven_id': self.ven_id,
- 'reports': reports}
- service = 'EiReport'
- message = self._create_message('oadrRegisterReport', **payload)
- response_type, response_payload = await self._perform_request(service, message)
- # Handle the subscriptions that the VTN is interested in.
- if 'report_requests' in response_payload:
- for report_request in response_payload['report_requests']:
- await self.create_report(report_request)
- message_type = 'oadrCreatedReport'
- message_payload = {}
- return message_type, message_payload
- async def create_report(self, report_request):
- """
- Add the requested reports to the reporting mechanism.
- This is called when the VTN requests reports from us.
- :param report_request dict: The oadrReportRequest dict from the VTN.
- """
- # Get the relevant variables from the report requests
- report_request_id = report_request['report_request_id']
- report_specifier_id = report_request['report_specifier']['report_specifier_id']
- report_back_duration = report_request['report_specifier'].get('report_back_duration')
- granularity = report_request['report_specifier']['granularity']
- # Check if this report actually exists
- report = utils.find_by(self.reports, 'report_specifier_id', report_specifier_id)
- if not report:
- logger.error(f"A non-existant report with report_specifier_id "
- f"{report_specifier_id} was requested.")
- return False
- # Check and collect the requested r_ids for this report
- requested_r_ids = []
- for specifier_payload in report_request['report_specifier']['specifier_payloads']:
- r_id = specifier_payload['r_id']
- # Check if the requested r_id actually exists
- rd = utils.find_by(report.report_descriptions, 'r_id', r_id)
- if not rd:
- logger.error(f"A non-existant report with r_id {r_id} "
- f"inside report with report_specifier_id {report_specifier_id} "
- f"was requested.")
- continue
- # Check if the requested measurement exists and if the correct unit is requested
- if 'measurement' in specifier_payload:
- measurement = specifier_payload['measurement']
- if measurement['description'] != rd.measurement.description:
- logger.error(f"A non-matching measurement description for report with "
- f"report_request_id {report_request_id} and r_id {r_id} was given "
- f"by the VTN. Offered: {rd.measurement.description}, "
- f"requested: {measurement['description']}")
- continue
- if measurement['unit'] != rd.measurement.unit:
- logger.error(f"A non-matching measurement unit for report with "
- f"report_request_id {report_request_id} and r_id {r_id} was given "
- f"by the VTN. Offered: {rd.measurement.unit}, "
- f"requested: {measurement['unit']}")
- continue
- if granularity is not None:
- if not rd.sampling_rate.min_period <= granularity <= rd.sampling_rate.max_period:
- logger.error(f"An invalid sampling rate {granularity} was requested for report "
- f"with report_specifier_id {report_specifier_id} and r_id {r_id}. "
- f"The offered sampling rate was between "
- f"{rd.sampling_rate.min_period} and "
- f"{rd.sampling_rate.max_period}")
- continue
- else:
- # If no granularity is specified, set it to the lowest sampling rate.
- granularity = rd.sampling_rate.max_period
- requested_r_ids.append(r_id)
- callback = partial(self.update_report, report_request_id=report_request_id)
- reporting_interval = report_back_duration or granularity
- job = self.scheduler.add_job(func=callback,
- trigger='cron',
- **utils.cron_config(reporting_interval))
- self.report_requests.append({'report_request_id': report_request_id,
- 'report_specifier_id': report_specifier_id,
- 'report_back_duration': report_back_duration,
- 'r_ids': requested_r_ids,
- 'granularity': granularity,
- 'job': job})
- async def create_single_report(self, report_request):
- """
- Create a single report in response to a request from the VTN.
- """
- async def update_report(self, report_request_id):
- """
- Call the previously registered report callback and send the result as a message to the VTN.
- """
- logger.debug(f"Running update_report for {report_request_id}")
- report_request = utils.find_by(self.report_requests, 'report_request_id', report_request_id)
- granularity = report_request['granularity']
- report_back_duration = report_request['report_back_duration']
- report_specifier_id = report_request['report_specifier_id']
- report = utils.find_by(self.reports, 'report_specifier_id', report_specifier_id)
- data_collection_mode = report.data_collection_mode
- if report_request_id in self.incomplete_reports:
- logger.debug("We were already compiling this report")
- outgoing_report = self.incomplete_reports[report_request_id]
- else:
- logger.debug("There is no report in progress")
- outgoing_report = objects.Report(report_request_id=report_request_id,
- report_specifier_id=report.report_specifier_id,
- report_name=report.report_name,
- intervals=[])
- intervals = outgoing_report.intervals or []
- if data_collection_mode == 'full':
- if report_back_duration is None:
- report_back_duration = granularity
- date_to = datetime.now(timezone.utc)
- date_from = date_to - max(report_back_duration, granularity)
- for r_id in report_request['r_ids']:
- report_callback = self.report_callbacks[(report_specifier_id, r_id)]
- result = report_callback(date_from=date_from,
- date_to=date_to,
- sampling_interval=granularity)
- if asyncio.iscoroutine(result):
- result = await result
- for dt, value in result:
- report_payload = objects.ReportPayload(r_id=r_id, value=value)
- intervals.append(objects.ReportInterval(dtstart=dt,
- report_payload=report_payload))
- else:
- for r_id in report_request['r_ids']:
- report_callback = self.report_callbacks[(report_specifier_id, r_id)]
- result = report_callback()
- if asyncio.iscoroutine(result):
- result = await result
- if isinstance(result, (int, float)):
- result = [(datetime.now(timezone.utc), result)]
- for dt, value in result:
- logger.info(f"Adding {dt}, {value} to report")
- report_payload = objects.ReportPayload(r_id=r_id, value=value)
- intervals.append(objects.ReportInterval(dtstart=dt,
- report_payload=report_payload))
- outgoing_report.intervals = intervals
- logger.info(f"The number of intervals in the report is now {len(outgoing_report.intervals)}")
- # Figure out if the report is complete after this sampling
- if data_collection_mode == 'incremental' and report_back_duration is not None\
- and report_back_duration > granularity:
- report_interval = report_back_duration.total_seconds()
- sampling_interval = granularity.total_seconds()
- expected_len = len(report_request['r_ids']) * int(report_interval / sampling_interval)
- if len(outgoing_report.intervals) == expected_len:
- logger.info("The report is now complete with all the values. Will queue for sending.")
- if self.allow_jitter:
- delay = random.uniform(0, min(30, report_interval / 2))
- self.loop.create_task(utils.delayed_call(func=self.pending_reports.put(outgoing_report),
- delay=delay))
- else:
- await self.pending_reports.put(self.incomplete_reports.pop(report_request_id))
- else:
- logger.debug("The report is not yet complete, will hold until it is.")
- self.incomplete_reports[report_request_id] = outgoing_report
- else:
- logger.info("Report will be sent now.")
- if self.allow_jitter:
- delay = random.uniform(0, min(30, granularity.total_seconds() / 2))
- self.loop.create_task(utils.delayed_call(func=self.pending_reports.put(outgoing_report),
- delay=delay))
- else:
- await self.pending_reports.put(outgoing_report)
- async def cancel_report(self, payload):
- """
- Cancel this report.
- """
- async def _report_queue_worker(self):
- """
- A Queue worker that pushes out the pending reports.
- """
- while True:
- report = await self.pending_reports.get()
- service = 'EiReport'
- message = self._create_message('oadrUpdateReport', reports=[report])
- try:
- response_type, response_payload = await self._perform_request(service, message)
- except Exception as err:
- logger.error(f"Unable to send the report to the VTN. Error: {err}")
- else:
- if 'cancel_report' in response_payload:
- await self.cancel_report(response_payload['cancel_report'])
- ###########################################################################
- # #
- # PLACEHOLDER #
- # #
- ###########################################################################
- async def on_event(self, event):
- """
- Placeholder for the on_event handler.
- """
- logger.warning("You should implement your own on_event handler. This handler receives "
- "an Event dict and should return either 'optIn' or 'optOut' based on your "
- "choice. Will opt out of the event for now.")
- return 'optOut'
- async def on_update_event(self, event):
- """
- Placeholder for the on_update_event handler.
- """
- logger.warning("You should implement your own on_update_event handler. This handler receives "
- "an Event dict and should return either 'optIn' or 'optOut' based on your "
- "choice. Will re-use the previous opt status for this event_id for now")
- if event['event_descriptor']['event_id'] in self.events:
- return self.responded_events['event_id']
- ###########################################################################
- # #
- # LOW LEVEL #
- # #
- ###########################################################################
- async def _perform_request(self, service, message):
- await self._ensure_client_session()
- logger.debug(f"Client is sending {message}")
- url = f"{self.vtn_url}/{service}"
- try:
- async with self.client_session.post(url, data=message) as req:
- content = await req.read()
- if req.status != HTTPStatus.OK:
- logger.warning(f"Non-OK status {req.status} when performing a request to {url} "
- f"with data {message}: {req.status} {content.decode('utf-8')}")
- return None, {}
- logger.debug(content.decode('utf-8'))
- except aiohttp.client_exceptions.ClientConnectorError as err:
- # Could not connect to server
- logger.error(f"Could not connect to server with URL {self.vtn_url}:")
- logger.error(f"{err.__class__.__name__}: {str(err)}")
- return None, {}
- except Exception as err:
- logger.error(f"Request error {err.__class__.__name__}:{err}")
- return None, {}
- try:
- tree = validate_xml_schema(content)
- if self.vtn_fingerprint:
- validate_xml_signature(tree)
- message_type, message_payload = parse_message(content)
- except XMLSyntaxError as err:
- logger.warning(f"Incoming message did not pass XML schema validation: {err}")
- return None, {}
- except errors.FingerprintMismatch as err:
- logger.warning(err)
- return None, {}
- except InvalidSignature:
- logger.warning("Incoming message had invalid signature, ignoring.")
- return None, {}
- except Exception as err:
- logger.error(f"The incoming message could not be parsed or validated: {err}")
- return None, {}
- if message_payload['response']['response_code'] != 200:
- logger.warning("We got a non-OK OpenADR response from the server: "
- f"{message_payload['response']['response_code']}: "
- f"{message_payload['response']['response_description']}")
- return message_type, message_payload
- async def _on_event(self, message):
- logger.debug("The VEN received an event")
- events = message['events']
- try:
- results = []
- for event in message['events']:
- event_id = event['event_descriptor']['event_id']
- event_status = event['event_descriptor']['event_status']
- if event_id in self.responded_events:
- result = self.on_update_event(event)
- else:
- result = self.on_event(event)
- if asyncio.iscoroutine(result):
- result = await result
- results.append(result)
- if event_status == 'completed':
- self.responded_events.pop(event_id)
- else:
- self.responded_events[event_id] = result
- for i, result in enumerate(results):
- if result not in ('optIn', 'optOut'):
- logger.error("Your on_event or on_update_event handler must return 'optIn' or 'optOut'; "
- f"you supplied {result}. Please fix your on_event handler.")
- results[i] = 'optOut'
- except Exception as err:
- logger.error("Your on_event handler encountered an error. Will Opt Out of the event. "
- f"The error was {err.__class__.__name__}: {str(err)}")
- results = ['optOut'] * len(events)
- if len(events) == 1:
- logger.debug(f"Now responding with {results[0]}")
- else:
- logger.debug(f"Responding to multiple events: {results}.")
- event_responses = [{'response_code': 200,
- 'response_description': 'OK',
- 'opt_type': results[i],
- 'request_id': message['request_id'],
- 'modification_number': 1,
- 'event_id': events[i]['event_descriptor']['event_id']}
- for i, event in enumerate(events)]
- response = {'response_code': 200,
- 'response_description': 'OK',
- 'request_id': message['request_id']}
- message = self._create_message('oadrCreatedEvent',
- response=response,
- event_responses=event_responses,
- ven_id=self.ven_id)
- service = 'EiEvent'
- response_type, response_payload = await self._perform_request(service, message)
- logger.info(response_type, response_payload)
- async def _poll(self):
- logger.debug("Now polling for new messages")
- response_type, response_payload = await self.poll()
- if response_type is None:
- return
- if response_type == 'oadrResponse':
- logger.debug("No events or reports available")
- return
- if response_type == 'oadrRequestReregistration':
- logger.info("The VTN required us to re-register. Calling the registration procedure.")
- await self.create_party_registration()
- if response_type == 'oadrDistributeEvent':
- if len(response_payload['events']) > 0:
- await self._on_event(response_payload)
- elif response_type == 'oadrUpdateReport':
- await self._on_report(response_payload)
- elif response_type == 'oadrCreateReport':
- if 'report_requests' in response_payload:
- for report_request in response_payload['report_requests']:
- await self.create_report(report_request)
- else:
- logger.warning(f"No handler implemented for incoming message "
- f"of type {response_type}, ignoring.")
- # Immediately poll again, because there might be more messages
- await self._poll()
- async def _ensure_client_session(self):
- if not self.client_session:
- if self.cert_path:
- ssl_context = ssl.create_default_context(cafile=self.ca_file,
- purpose=ssl.Purpose.CLIENT_AUTH)
- ssl_context.load_cert_chain(self.cert_path, self.key_path, self.passphrase)
- ssl_context.check_hostname = False
- connector = aiohttp.TCPConnector(ssl=ssl_context)
- self.client_session = aiohttp.ClientSession(connector=connector)
- else:
- self.client_session = aiohttp.ClientSession()
|