Syndication - update keys from match document

1 files changed, 38 insertions, 4 deletions

diff --git a/ishtar_common/libreoffice.py b/ishtar_common/libreoffice.py
index 6337675fe..482364a73 100644
--- a/ishtar_common/libreoffice.py
+++ b/ishtar_common/libreoffice.py
@@ -74,12 +74,22 @@ class UnoCalc(UnoClient):
     def create_calc(self):
         return self.create_document('scalc')
 
+    def open_calc(self, filename):
+        return self.get_document(filename)
+
     def save_calc(self, calc, filename):
         url = "file://{}".format(filename)
         args = (PropertyValue(Name="FilterName", Value="Calc8"),)
         calc.storeToURL(url, args)
 
-    def get_sheet(self, calc, sheet_index, name=None):
+    def get_sheet_number(self, calc):
+        sheets = calc.getSheets()
+        return sheets.Count
+
+    def get_sheet(self, calc, sheet_index):
+        return calc.getSheets().getByIndex(sheet_index)
+
+    def get_or_create_sheet(self, calc, sheet_index, name=None):
         sheets = calc.getSheets()
         while sheets.Count < (sheet_index + 1):
             if name and sheets.Count == sheet_index:
@@ -89,6 +99,30 @@ class UnoCalc(UnoClient):
             sheets.insertNewByName(sheet_name, sheets.Count)
         return calc.getSheets().getByIndex(sheet_index)
 
+    def sheet_get_last_row_number(self, sheet):
+        cursor = sheet.createCursor()
+        cursor.gotoEndOfUsedArea(False)
+        return cursor.getRangeAddress().EndRow + 1
+
+    def sheet_get_data(self, sheet, last_column=None):
+        if not last_column:
+            # assume the first line is header with no empty columns
+            last_column = 0
+            has_data = True
+            while has_data:
+                has_data = sheet.getCellByPosition(last_column, 0).getString()
+                if not has_data:
+                    break
+                last_column += 1
+        last_row = self.sheet_get_last_row_number(sheet)
+        data = []
+        for row in range(last_row):
+            data.append([])
+            for col in range(last_column):
+                value = sheet.getCellByPosition(col, row).getString()
+                data[-1].append(value)
+        return data
+
     def create_list(self, sheet, col, start_row, title, items):
         items = [title] + items
         row_idx = 0
@@ -147,11 +181,11 @@ class UnoCalc(UnoClient):
         if not lst:  # no list
             return
         calc = self.create_calc()
-        lst_sheet = self.get_sheet(calc, 1, "List types")
+        lst_sheet = self.get_or_create_sheet(calc, 1, "List types")
 
         lst_col, lst_row = 0, 0
         end_row = self.create_list(lst_sheet, lst_col, lst_row, title, lst)
-        main_sheet = self.get_sheet(calc, 0)
+        main_sheet = self.get_or_create_sheet(calc, 0)
         self.set_cell_validation_list(main_sheet, 0, 0, 200, lst_sheet, lst_col,
                                       [lst_row + 1, end_row])
         self.save_calc(calc, "/tmp/test.ods")
@@ -162,7 +196,7 @@ class UnoCalc(UnoClient):
         propval.Value = True
         filename = "/tmp/main.ods"
         calc = self.get_document(filename, (propval,))
-        sheet = self.get_sheet(calc, 0)
+        sheet = self.get_or_create_sheet(calc, 0)
         validation = sheet.getCellByPosition(0, 0).Validation
         for k in dir(validation):
             if k.startswith("get"):