From bc0135900444cb45e6ff0b583f770b55b033fc0f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Nicolas=20H=C3=B6ning?= <iam@nicolashoening.de>
Date: Wed, 19 May 2021 13:02:57 +0200
Subject: [PATCH] add delimiter and decimal options to dev-add beliefs CLI
 command (#134)

---
 .../data/scripts/cli_tasks/data_add.py         | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/flexmeasures/data/scripts/cli_tasks/data_add.py b/flexmeasures/data/scripts/cli_tasks/data_add.py
index d1e01ec25..729d95103 100644
--- a/flexmeasures/data/scripts/cli_tasks/data_add.py
+++ b/flexmeasures/data/scripts/cli_tasks/data_add.py
@@ -287,6 +287,20 @@ def add_initial_structure():
     type=int,
     help="Column number with values (1 is 2nd column, the default)",
 )
+@click.option(
+    "--decimal",
+    required=False,
+    default=".",
+    type=str,
+    help="[For csv files] decimal character, e.g. '.' for 10.5",
+)
+@click.option(
+    "--delimiter",
+    required=True,
+    type=str,
+    default=",",
+    help="[For csv files] Character to delimit columns per row, defaults to comma",
+)
 @click.option(
     "--sheet_number",
     required=False,
@@ -304,6 +318,8 @@ def add_beliefs(
     nrows: Optional[int] = None,
     datecol: int = 0,
     valuecol: int = 1,
+    decimal: str = ".",
+    delimiter: str = ",",
     sheet_number: Optional[int] = None,
 ):
     """Add sensor data from a csv file (also accepts xls or xlsx).
@@ -340,6 +356,8 @@ def add_beliefs(
     kwargs = dict()
     if file.split(".")[-1].lower() == "csv":
         kwargs["infer_datetime_format"] = True
+        kwargs["decimal"] = decimal
+        kwargs["delimiter"] = delimiter
     if sheet_number is not None:
         kwargs["sheet_name"] = sheet_number
     if horizon is not None: