maisaedu-utilities-prefect 1.2.66__tar.gz → 1.2.68__tar.gz

{maisaedu_utilities_prefect-1.2.66 → maisaedu_utilities_prefect-1.2.68}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: maisaedu_utilities_prefect
-Version: 1.2.66
+Version: 1.2.68
 Summary: Utilities for interaction with Prefect, for +A Education
 Home-page: UNKNOWN
 Author: A+ Educação

{maisaedu_utilities_prefect-1.2.66 → maisaedu_utilities_prefect-1.2.68}/maisaedu_utilities_prefect/one_drive/document_handling.py

@@ -31,34 +31,63 @@ def read_docs(documents):
 
     return df
 
-def __get_shared_files_info(headers):
-    url = "/me/drive/sharedWithMe"
+def __get_shared_files_info(headers, query_string):
+    url = "/search/query"
 
-    response_file_info = requests.get(
+    payload = {
+        "requests": [
+            {
+                "entityTypes": ["driveItem"],
+                "query": {
+                    "queryString": f"{query_string}"
+                },
+                "from": 0,
+                "size": 10
+            }
+        ]
+    }
+
+    response_file_info = requests.post(
         GRAPH_API_ENDPOINT + url,
         headers=headers,
+        json=payload
     )
 
-    values = response_file_info.json()["value"]
+    values = response_file_info.json()["value"][0]["hitsContainers"][0]["hits"]
 
     return values
 
+def get_shared_documents_by_query_search(query_string):
+    token = get_secrets()
+
+    values = __get_shared_files_info({"Authorization": f'Bearer {token["access_token"]}'}, query_string)
+    shared_documents = []
+
+    for value in values:
+            shared_documents.append({
+                "id": value["resource"]["sharepointIds"]["listItemUniqueId"],
+                "name": value["name"],
+                "url": value["webUrl"]
+            })
+
+    return shared_documents
+
 def __get_shared_info(headers, file_id, file_url = None):
-    values = __get_shared_files_info(headers)
+    values = __get_shared_files_info(headers, file_id)
     shared_documents = []
 
     try:
         for value in values:
-            listItemUniqueId = value["remoteItem"]["sharepointIds"]["listItemUniqueId"]
-            web_url = value["webUrl"]
+            listItemUniqueId = value["resource"]["parentReference"]["sharepointIds"]["listItemUniqueId"]
+            web_url = value["resource"]["webUrl"]
             shared_documents.append({
                 "id": listItemUniqueId,
-                "name": value["name"],
+                "name": value["resource"]["name"],
                 "web_url": web_url
             })
             if listItemUniqueId == file_id.lower() or web_url == file_url:
-                drive_id = value["remoteItem"]["parentReference"]["driveId"]
-                item_id = value["id"]
+                drive_id = value["resource"]["parentReference"]["driveId"]
+                item_id = value["resource"]["id"]
                 unique_id = listItemUniqueId
         return drive_id, item_id, unique_id
     except UnboundLocalError:
@@ -139,21 +168,6 @@ def get_documents_info(files_info):
 
     return documents_info
 
-def get_all_shared_documents():
-    token = get_secrets()
-
-    values = __get_shared_files_info({"Authorization": f'Bearer {token["access_token"]}'})
-    shared_documents = []
-
-    for value in values:
-            shared_documents.append({
-                "id": value["remoteItem"]["sharepointIds"]["listItemUniqueId"],
-                "name": value["name"],
-                "url": value["webUrl"]
-            })
-
-    return shared_documents
-
 def delete_docs(documents_names):
     for document_name in documents_names:
         os.remove(document_name)

{maisaedu_utilities_prefect-1.2.66 → maisaedu_utilities_prefect-1.2.68}/maisaedu_utilities_prefect.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: maisaedu-utilities-prefect
-Version: 1.2.66
+Version: 1.2.68
 Summary: Utilities for interaction with Prefect, for +A Education
 Home-page: UNKNOWN
 Author: A+ Educação

{maisaedu_utilities_prefect-1.2.66 → maisaedu_utilities_prefect-1.2.68}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="maisaedu_utilities_prefect",
-    version="1.2.66",
+    version="1.2.68",
     description="Utilities for interaction with Prefect, for +A Education",
     license="MIT License",
     author="A+ Educação",