Main brez modelov

swagger_server/controllers/oss_controller.py

@@ -1,21 +1,6 @@
-import connexion
-import six
-
-from swagger_server.models.terminoloski_kandidat import TerminoloskiKandidat  # noqa: E501
+from swagger_server.utils import db_utils
 from swagger_server import util
-
-
-def get_conllu(id):  # noqa: E501
-    """Vrne CoNNL-U po id-ju datoteke
-
-     # noqa: E501
-
-    :param id: 
-    :type id: int
-
-    :rtype: str
-    """
-    return 'do some magic!'
+from flask import send_file
 
 
 def get_conllus(leta, vrste, kljucnebesede, cerifpodrocja):  # noqa: E501
@@ -34,10 +19,16 @@ def get_conllus(leta, vrste, kljucnebesede, cerifpodrocja):  # noqa: E501
 
     :rtype: List[str]
     """
-    return 'do some magic!'
+    if not kljucnebesede:
+        return "Manjkajo kljucne besede", 400
+    #zaenkrat ne potrebujemo te storitve
+    files = db_utils.get_files_by_udc(kljucnebesede)
+    if not files:
+        return 'Nobena datoteka ne ustreza iskalnemu pogoju', 404
+    return ' '.join(files), 200
 
 
-def get_extracted_words(leta, vrste, kljucnebesede, cerifpodrocja):  # noqa: E501
+def get_extracted_words(leta=None, vrste=None, kljucnebesede=None, udk=None):  # noqa: E501
     """Vrne terminloške kandidate glede na 
 
      # noqa: E501
@@ -53,21 +44,10 @@ def get_extracted_words(leta, vrste, kljucnebesede, cerifpodrocja):  # noqa: E50
 
     :rtype: List[TerminoloskiKandidat]
     """
-    return 'do some magic!'
+    files = db_utils.vrni_oss_terminoloske_kandidate(leta, vrste, kljucnebesede, udk)
+    return files, 200
 
 
-def get_file(id):  # noqa: E501
-    """Vrne binarni zapis v originalnem formatu po id-ju datoteke
-
-     # noqa: E501
-
-    :param id: 
-    :type id: int
-
-    :rtype: List[bytearray]
-    """
-    return 'do some magic!'
-
 
 def get_files(leta, vrste, kljucnebesede, cerifpodrocja):  # noqa: E501
     """Vrne seznam binarnih zapisov v originalnem formatu glede na iskalne pogoje
@@ -85,10 +65,16 @@ def get_files(leta, vrste, kljucnebesede, cerifpodrocja):  # noqa: E501
 
     :rtype: List[List[bytearray]]
     """
-    return 'do some magic!'
+    if not kljucnebesede:
+        return "Manjkajo kljucne besede", 400
+    #zaenkrat ne potrebujemo te storitve
+    files = db_utils.get_files_by_udc(kljucnebesede)
+    if not files:
+        return 'Nobena datoteka ne ustreza iskalnemu pogoju', 404
+    return ' '.join(files), 200
 
 
-def get_number_texts(leta, vrste, kljucnebesede, cerifpodrocja):  # noqa: E501
+def get_number_texts(leta=None, vrste=None, kljucnebesede=None, udk=None):  # noqa: E501
     """Vrne število besedil glede na iskalne pogoje
 
      # noqa: E501
@@ -100,11 +86,15 @@ def get_number_texts(leta, vrste, kljucnebesede, cerifpodrocja):  # noqa: E501
     :param kljucnebesede: 
     :type kljucnebesede: List[str]
     :param cerifpodrocja: 
-    :type cerifpodrocja: List[int]
+    :type udc: List[int]
 
     :rtype: int
     """
-    return 'do some magic!'
+    #if not kljucnebesede:
+    #    return "Manjkajo kljucne besede", 400
+
+    files = db_utils.vrni_oss_dokumente(leta, vrste, kljucnebesede, udk)
+    return len(files), 200
 
 
 def get_texts(leta, vrste, kljucnebesede, cerifpodrocja):  # noqa: E501
@@ -123,17 +113,60 @@ def get_texts(leta, vrste, kljucnebesede, cerifpodrocja):  # noqa: E501
 
     :rtype: List[str]
     """
-    return 'do some magic!'
+    if not kljucnebesede:
+        return "Manjkajo kljucne besede", 400
+    #zaenkrat ne potrebujemo te storitve
+    files = db_utils.get_files_by_udc(kljucnebesede)
+    if not files:
+        return 'Nobena datoteka ne ustreza iskalnemu pogoju', 404
+    return ' '.join(files), 200
 
 
-def oss_besedilo_po_id_get(id):  # noqa: E501
+def get_conllu(file_id):  # noqa: E501
+    """Vrne CoNNL-U po id-ju datoteke
+
+     # noqa: E501
+
+    :param file_id:
+    :type file_id: int
+
+    :rtype: str
+    """
+    try:
+        return send_file(util.get_conllu_file_path_by_id(file_id), download_name=f'{file_id}.conllu')
+    except FileNotFoundError as e:
+        return "The conllu with this ID doesn't exist.", 404
+
+
+def get_file(file_id):  # noqa: E501
+    """Vrne binarni zapis v originalnem formatu po id-ju datoteke
+
+     # noqa: E501
+
+    :param file_id:
+    :type file_id: int
+
+    :rtype: List[bytearray]
+    """
+    try:
+        return send_file(util.get_original_file_path_by_id(file_id), download_name=f'{file_id}.xml')
+    except FileNotFoundError as e:
+        return "The file with this ID doesn't exist.", 404
+
+
+def oss_besedilo_po_id_get(file_id):  # noqa: E501
     """Vrne besedilo po id-ju datoteke
 
      # noqa: E501
 
-    :param id: 
-    :type id: int
+    :param file_id:
+    :type file_id: int
 
     :rtype: str
     """
-    return 'do some magic!'
+    try:
+        f = util.get_original_file_path_by_id(file_id)
+        print(f) # for debugging purposes on the server, delete this later
+        return send_file(util.get_original_file_path_by_id(file_id), download_name=f'{file_id}.xml')
+    except FileNotFoundError as e:
+        return "The file with this ID doesn't exist.", 404