Allow fetching of tibber rates in the inverter script

Author: felixschndr

inverter

@@ -5,7 +5,7 @@ SCRIPT_DIR="$(dirname "$(readlink -f "$0")")"
 source "${SCRIPT_DIR}"/.venv/bin/activate
 cd "${SCRIPT_DIR}"/source || exit 1
 
-read -r -d '' setup_code <<'EOF'
+read -r -d '' inverter_setup_code <<'EOF'
 from goodwe.et import OperationMode
 from inverter import Inverter
 
@@ -14,20 +14,21 @@ inverter.log.info("Inverter is manually controlled by a user")
 EOF
 
 
-if [[ ! ${1} =~ ^(--status|--mode)$ ]]; then
+if [[ ! ${1} =~ ^(--status|--mode|--tibber)$ ]]; then
 	echo "This script allows you to interact with the inverter."
 	echo ""
 	echo "You have to supply one of the following arguments:"
 	echo " --status                    Get the current state of charge and operation mode of the inverter"
 	echo " --mode {GENERAL,ECO_CHARGE} Set the operation mode of the inverter"
+	echo " --tibber                    Fetches, displays the tibber rates and writes them to the database"
 	echo ""
-	echo "It uses the hostname defined in the .env file"
+	echo "It uses the all the variables defined in the .env file"
 	exit 1
 fi
 
 if [[ ${1} == "--status" ]]; then
 	python3 - << EOF
-${setup_code}
+${inverter_setup_code}
 try:
 	mode = inverter.get_operation_mode().name
 	soc = inverter.get_state_of_charge()
@@ -36,7 +37,7 @@ except Exception as e:
 	inverter.log.exception(e)
 	exit(1)
 EOF
-exit $?
+	exit $?
 fi
 
 if [[ ${1} == "--mode" ]]; then
@@ -46,7 +47,7 @@ if [[ ${1} == "--mode" ]]; then
 	fi
 
 	python3 - << EOF
-${setup_code}
+${inverter_setup_code}
 try:
 	last_mode = inverter.get_operation_mode().name
 	inverter.set_operation_mode(OperationMode.${2})
@@ -55,5 +56,21 @@ except Exception as e:
 	inverter.log.exception(e)
 	exit(1)
 EOF
-exit $?
+	exit $?
+fi
+
+if [[ ${1} == "--tibber" ]]; then
+	python3 - << EOF
+from tibber_api_handler import TibberAPIHandler
+try:
+	tibber_api_handler = TibberAPIHandler()
+	api_result = tibber_api_handler._fetch_upcoming_prices_from_api()
+	all_energy_rates = tibber_api_handler._extract_energy_rates_from_api_response(api_result)
+	tibber_api_handler._remove_energy_rates_from_the_past(all_energy_rates)
+	tibber_api_handler.write_energy_rates_to_database(all_energy_rates)
+except Exception as e:
+	tibber_api_handler.log.exception(e)
+	exit(1)
+EOF
+	exit $?
 fi