Merge pull request #135 from IndicoDataSolutions/add_num_pages_to_report

adding two attributes to doc reports
IndicoDataSolutions · Feb 2, 2022 · f5dcbac · f5dcbac
2 parents a168c2a + f219c9e
commit f5dcbac
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/indico/queries/document_report.py b/indico/queries/document_report.py
@@ -35,6 +35,8 @@ class GetDocumentReport(PagedRequest):
               inputFiles{
                 filename
                 submissionId
+                fileSize
+                numPages
               }
             }
             pageInfo{

diff --git a/indico/types/document_report.py b/indico/types/document_report.py
@@ -6,6 +6,8 @@
 class InputFile(BaseType):
     filename: str
     submission_id: int
+    num_pages: int
+    file_size: int
 
 
 class DocumentReport(BaseType):

diff --git a/tests/integration/queries/test_document_report.py b/tests/integration/queries/test_document_report.py
@@ -6,21 +6,30 @@
 from indico.types.document_report import DocumentReport
 from indico.filters import DocumentReportFilter
 
+
 def test_fetch_docs(indico):
     client = IndicoClient()
 
     document_report: List[DocumentReport] = client.call(GetDocumentReport())
     assert document_report is not None
     assert len(document_report) > 0
+    assert isinstance(document_report[0].input_files[0].num_pages, int)
+    assert isinstance(document_report[0].input_files[0].file_size, int)
+
 
 def test_fetch_docs_limit(indico):
     client = IndicoClient()
-    filter_opts = DocumentReportFilter(created_at_start_date= datetime(2021,7,1), created_at_end_date = datetime.now())
-    document_report: List[DocumentReport] = client.call(GetDocumentReport(filters=filter_opts, limit=5))
+    filter_opts = DocumentReportFilter(
+        created_at_start_date=datetime(2021, 7, 1), created_at_end_date=datetime.now()
+    )
+    document_report: List[DocumentReport] = client.call(
+        GetDocumentReport(filters=filter_opts, limit=5)
+    )
     assert document_report is not None
     assert len(document_report) > 0
     assert len(document_report) <= 5
 
+
 def test_pagination(indico):
     client = IndicoClient()
     document_report: List[DocumentReport] = []