Merge remote-tracking branch 'origin/main'

# Conflicts:
#	Dockerfile
This commit is contained in:
Jan-Ole Hübner 2024-08-18 00:06:51 +02:00
commit 610cdd6a65
2 changed files with 54 additions and 62 deletions

View file

@ -1,35 +0,0 @@
import os
import sys
import tibber
from influxdb_client import InfluxDBClient
from influxdb_client.client.write_api import SYNCHRONOUS
from DataPoints import Price
import logging
TOKEN = os.getenv('TOKEN', '')
TIBBERTOKEN = os.getenv('TIBBERTOKEN', '')
URL = os.getenv('URL', "")
BUCKET = os.getenv('BUCKET', "tibber")
ORG = os.getenv('ORG', "Default")
# Logging configuration
logger = logging.getLogger("TibberInflux")
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s')
ch = logging.StreamHandler(sys.stdout)
ch.setFormatter(formatter)
logger.addHandler(ch)
logger.setLevel(logging.INFO)
__version__ = "v0.3.0"
logger.info(__version__)
client = InfluxDBClient(url=URL, token=TOKEN, org=ORG)
write_api = client.write_api(write_options=SYNCHRONOUS)
query_api = client.query_api()
account = tibber.Account(TIBBERTOKEN)
home = account.homes[0]
p = Price(home.current_subscription.price_info.current).get_datapoint()
write_api.write(record=p, bucket=BUCKET)
logger.info(vars(p))

View file

@ -1,13 +1,13 @@
import asyncio
import os import os
import signal import signal
import sys import sys
import logging
import tibber import tibber
from influxdb_client import InfluxDBClient from influxdb_client import InfluxDBClient
from influxdb_client.client.write_api import SYNCHRONOUS from influxdb_client.client.write_api import SYNCHRONOUS
from DataPoints import Pulse from DataPoints import Pulse, Price
import logging
TOKEN = os.getenv('TOKEN', '') TOKEN = os.getenv('TOKEN', '')
TIBBERTOKEN = os.getenv('TIBBERTOKEN', '') TIBBERTOKEN = os.getenv('TIBBERTOKEN', '')
@ -17,25 +17,24 @@ ORG = os.getenv('ORG', "Default")
# Logging configuration # Logging configuration
logger = logging.getLogger("TibberInflux") logger = logging.getLogger("TibberInflux")
formatter = logging.Formatter( formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
'%(asctime)s - %(name)s - %(levelname)s - %(message)s')
ch = logging.StreamHandler(sys.stdout) ch = logging.StreamHandler(sys.stdout)
ch.setFormatter(formatter) ch.setFormatter(formatter)
logger.addHandler(ch) logger.addHandler(ch)
logger.setLevel(logging.INFO) logger.setLevel(logging.INFO)
__version__ = "v0.3.5_debug" __version__ = "v0.4.0"
logger.info(__version__) logger.info(__version__)
# Initialize InfluxDB client
logger.info("connecting to db...")
client = InfluxDBClient(url=URL, token=TOKEN, org=ORG) client = InfluxDBClient(url=URL, token=TOKEN, org=ORG)
write_api = client.write_api(write_options=SYNCHRONOUS) write_api = client.write_api(write_options=SYNCHRONOUS)
query_api = client.query_api() query_api = client.query_api()
logger.info("connecting to tibber...")
# Initialize Tibber account
account = tibber.Account(TIBBERTOKEN) account = tibber.Account(TIBBERTOKEN)
home = account.homes[0] home = account.homes[0]
logger.info("starting subscription...")
def _incoming(data): def _incoming(data):
try: try:
@ -46,32 +45,60 @@ def _incoming(data):
logger.exception("Error in _incoming():") logger.exception("Error in _incoming():")
raise e raise e
def timeout_handler(signum, frame): def timeout_handler(signum, frame):
raise TypeError("Timeout occurred") raise TypeError("Timeout occurred")
@home.event("live_measurement")
async def show_current_power(data): async def show_current_power(data):
signal.signal(signal.SIGALRM, timeout_handler)
signal.alarm(8) signal.alarm(8)
_incoming(data) _incoming(data)
def stop(home):
return False
try: async def periodic_task():
while True:
try:
# Fetch price info and write to InfluxDB
p = Price(home.current_subscription.price_info.current).get_datapoint()
write_api.write(record=p, bucket=BUCKET)
logger.info(vars(p))
except Exception as e:
logger.exception("Error in periodic_task():")
# Wait for the next run (e.g., every 60 seconds)
await asyncio.sleep(3)
async def main():
loop = asyncio.get_event_loop()
signal.signal(signal.SIGALRM, timeout_handler) signal.signal(signal.SIGALRM, timeout_handler)
signal.alarm(15) signal.alarm(15)
home.start_live_feed(user_agent="pulse.py/0.3.5",exit_condition=stop(home),retries=2,retry_interval=3.0)
except TypeError: try:
# Start the live feed in a separate task
home_feed_task = loop.create_task(home.start_live_feed(
user_agent="pulse.py/0.4.0",
exit_condition=lambda: False,
retries=2,
retry_interval=3.0,
callback=show_current_power
))
# Start the periodic task
periodic_task_task = loop.create_task(periodic_task())
# Wait for both tasks to complete
await asyncio.gather(home_feed_task, periodic_task_task)
except TypeError:
logger.exception("Timeout occurred while executing start_live_feed()") logger.exception("Timeout occurred while executing start_live_feed()")
client.close() client.close()
exit(1) sys.exit(1)
except Exception: except Exception:
logger.exception("Error in start_live_feed()") logger.exception("Error in start_live_feed()")
client.close() client.close()
exit(41) sys.exit(41)
if __name__ == "__main__":
asyncio.run(main())