Added WorkbooApplication, needs more work

Author: icgranger9

O365/excel.py

@@ -3,7 +3,7 @@
 Note: Support for workbooks stored in OneDrive Consumer platform is still not available.
 At this time, only the files stored in business platform is supported by Excel REST APIs.
 """
-
+import json
 import logging
 import datetime as dt
 from urllib.parse import quote
@@ -1833,3 +1833,72 @@ def add_named_range(self, name, reference, comment='', is_formula=False):
         if not response:
             return None
         return self.named_range_constructor(parent=self, **{self._cloud_data_key: response.json()})
+
+
+class WorkbookApplication(ApiComponent):
+    _endpoints = {
+        'get_workbookapplication': '/{id}/workbook/application',
+        'post_calculation': '/{id}/workbook/application/calculate'
+    }
+
+    def __init__(self, workbook_or_id, *, parent=None, con=None, **kwargs):
+        """
+        Create A WorkbookApplication representation
+
+        :param workbook_or_id: Either a workbook object, or the id of the workbook you want to interact with
+        :param parent: parent for this operation
+        :param Connection con: connection to use if no parent specified
+        """
+
+        if parent and con:
+            raise ValueError('Need a parent or a connection but not both')
+
+        self.con = parent.con if parent else con
+
+        # Assign the workbook id
+        if isinstance(workbook_or_id, WorkBook):
+            self.workbook_id.object_id.split(":").last()
+
+        else:
+            # Should probably do some more thorough testing
+            self.workbook_id = workbook_or_id
+
+        # Choose the main_resource passed in kwargs over parent main_resource
+        main_resource = kwargs.pop('main_resource', None) or (
+            getattr(parent, 'main_resource', None) if parent else None)
+
+        super().__init__(
+            protocol=parent.protocol if parent else kwargs.get('protocol'),
+            main_resource=main_resource)
+
+    def __str__(self):
+        return self.__repr__()
+
+    def __repr__(self):
+        return 'WorkbookApplication for Workbook: {}'.format(self.workbook_id or 'Not set')
+
+    def __bool__(self):
+        return self.workbook_id is not None
+
+    def get_workbookapplication(self):
+        """ Gets workbookApplication """
+        url = self.build_url(self._endpoints.get('get_workbookapplication').format(id=quote(self.workbook_id)))
+        headers = {
+
+        }
+        response = self.con.get(url, headers)
+        if not response:
+            return None
+        return response.json()
+
+    def run_calculations(self, calculation_type):
+        if calculation_type not in ["Recalculate", "Full", "FullRebuild"]:
+            raise ValueError("calculation type must be one of: Recalculate, Full, FullRebuild")
+
+        url = self.build_url(self._endpoints.get('get_workbookapplication').format(id=quote(self.workbook_id)))
+        body = json.dumps({"calculationType": calculation_type})
+
+        response = self.con.post(url, body)
+        if not response:
+            return False
+        return response.status_code == 200

tests/test_excel.py

@@ -0,0 +1,39 @@
+from O365 import Account, MSGraphProtocol, Connection
+from O365.excel import WorkbookApplication
+from tests.config import CLIENT_ID, CLIENT_SECRET
+
+
+class Workbook(object):
+    pass
+
+
+class TestWorkbookApplication:
+
+    def setup_class(self):
+        # credentials = ("client id", "client secret")
+        self.credentials = (CLIENT_ID, CLIENT_SECRET)
+        self.workbook_id = "B6EF05E2E8ABA433!646"
+        self.protocol = MSGraphProtocol()
+        con = Connection(self.credentials)
+        main_resource = "drive/items"
+
+        # Create Workbook Application
+        self.workbook_app = WorkbookApplication(workbook_or_id=self.workbook_id, con=con, protocol=self.protocol, main_resource=main_resource)
+
+    def teardown_class(self):
+        pass
+
+    def test_with_workbook(self):
+        # wb = Workbook()
+        pass
+
+    def test_with_parent(self):
+        pass
+
+    def test_get(self):
+        res = self.workbook_app.get_workbookapplication()
+        print(res)
+
+    def test_run_calculations(self):
+        res = self.workbook_app.run_calculations("Recalculate")
+        print(res)