OneDrive new actions

vgrem · vgrem · commit 1951be0666b5 · 2025-05-26T21:49:10.000+03:00
diff --git a/examples/directory/applications/has_application_perms.py b/examples/directory/applications/has_application_perms.py
@@ -22,7 +22,7 @@
 resource = client.service_principals.get_by_name("Microsoft Graph")
 app_role = "Place.Read.All"  # "Bookings.Read.All"
 result = resource.get_application_permissions(test_client_id).execute_query()
-#if app_role not in result.value:
+# if app_role not in result.value:
 if not any(role.value == app_role for role in result.value):
     print("Application permission '{0}' is not granted".format(app_role))
 else:
diff --git a/office365/directory/identitygovernance/entitlementmanagement/accesspackage/__init__.py b/office365/directory/identitygovernance/entitlementmanagement/accesspackage/__init__.py
diff --git a/office365/directory/identitygovernance/entitlementmanagement/accesspackage/catalog.py b/office365/directory/identitygovernance/entitlementmanagement/accesspackage/catalog.py
@@ -0,0 +1,10 @@
+from office365.entity import Entity
+
+
+class AccessPackageCatalog(Entity):
+    """In Microsoft Entra entitlement management, an access package catalog is a container for zero or more access
+    packages. Microsoft Entra entitlement management includes a built-in catalog named General.
+
+     An access package catalog might also have linked resources that are used in those access packages
+     to provide access. To view or change the membership of catalog-scoped roles, use the role assignments API
+     with the entitlement management RBAC provider."""
diff --git a/office365/directory/identitygovernance/entitlementmanagement/accesspackage/resource_request.py b/office365/directory/identitygovernance/entitlementmanagement/accesspackage/resource_request.py
@@ -0,0 +1,8 @@
+from office365.entity import Entity
+
+
+class AccessPackageResourceRequest(Entity):
+    """In Microsoft Entra entitlement management, an access package resource request is a request
+    to add a resource to a catalog so that the roles of the resource can be used in one or more of the catalog's
+    access packages, update a resource in a catalog to have different attribute requirements, or to remove a resource
+    from a catalog that is no longer needed by the access packages."""
diff --git a/office365/directory/identitygovernance/entitlementmanagement/entitlement_management.py b/office365/directory/identitygovernance/entitlementmanagement/entitlement_management.py
@@ -1,7 +1,24 @@
+from office365.directory.identitygovernance.entitlementmanagement.accesspackage.resource_request import (
+    AccessPackageResourceRequest,
+)
 from office365.entity import Entity
+from office365.entity_collection import EntityCollection
+from office365.runtime.paths.resource_path import ResourcePath
 
 
 class EntitlementManagement(Entity):
     """The entitlement management singleton is the container for entitlement management resources,
     including accessPackageCatalog, connectedOrganization, and entitlementManagementSettings.
     """
+
+    @property
+    def resource_requests(self):
+        """Represents a request to add or remove a resource to or from a catalog respectively."""
+        return self.properties.get(
+            "resourceRequests",
+            EntityCollection(
+                self.context,
+                AccessPackageResourceRequest,
+                ResourcePath("resourceRequests", self.resource_path),
+            ),
+        )
diff --git a/office365/directory/identitygovernance/workflow/custom_callout_extension.py b/office365/directory/identitygovernance/workflow/custom_callout_extension.py
@@ -0,0 +1,13 @@
+from typing import Optional
+
+from office365.entity import Entity
+
+
+class CustomCalloutExtension(Entity):
+    """An abstract type that defines the configuration for apps that can extend the customer's identity flows."""
+
+    @property
+    def display_name(self):
+        # type: () -> Optional[str]
+        """Display name for the customCalloutExtension object."""
+        return self.properties.get("displayName", None)
diff --git a/office365/directory/users/user.py b/office365/directory/users/user.py
@@ -233,6 +233,59 @@ def change_password(self, current_password, new_password):
         self.context.add_query(qry)
         return self
 
+    def follow_site(self, site):
+        # type: (str|Site) -> None
+        """follow a site"""
+
+        def _follow_site(site_id):
+            # type: (str) -> None
+            payload = {
+                "value": [
+                    {
+                        "id": site_id,
+                    }
+                ]
+            }
+            query = ServiceOperationQuery(self.followed_sites, "add", None, payload)
+            self.context.add_query(query)
+
+        if isinstance(site, Site):
+
+            def _user_loaded():
+                _follow_site(site.id)
+
+            site.ensure_property("id", _user_loaded)
+        else:
+            _follow_site(site)
+        return self
+
+    def unfollow_site(self, site):
+        # type: (str|Site) -> None
+        """Unfollow a user's site or multiple sites."""
+
+        def _unfollow_site(site_id):
+            # type: (str) -> None
+            payload = {
+                "value": [
+                    {
+                        "id": site_id,
+                    }
+                ]
+            }
+            query = ServiceOperationQuery(self.followed_sites, "remove", None, payload)
+            self.context.add_query(query)
+
+        if isinstance(site, Site):
+
+            def _user_loaded():
+                _unfollow_site(site.id)
+
+            site.ensure_property("id", _user_loaded)
+        else:
+            _unfollow_site(site)
+
+        return self
+
     def get_mail_tips(self, email_addresses, mail_tips_options=None):
         """Get the MailTips of one or more recipients as available to the signed-in user.
         :param list[str] email_addresses: A collection of SMTP addresses of recipients to get MailTips for.
diff --git a/office365/graph_client.py b/office365/graph_client.py
@@ -527,3 +527,7 @@ def security(self):
     def schema_extensions(self):
         """Get a list of schemaExtension objects in your tenant"""
         return EntityCollection(self, SchemaExtension, ResourcePath("schemaExtensions"))
+
+    @property
+    def tenant_name(self):
+        return self._tenant
diff --git a/office365/onedrive/internal/paths/site.py b/office365/onedrive/internal/paths/site.py
@@ -1,21 +1,17 @@
-from office365.runtime.compat import is_absolute_url, urlparse
+from office365.runtime.compat import urlparse
 from office365.runtime.paths.resource_path import ResourcePath
 from office365.runtime.paths.v4.entity import EntityPath
 
 
 class SitePath(EntityPath):
     """Resource path for addressing Site resource"""
 
-    @property
-    def segment(self):
-        if is_absolute_url(self._key):
-            url_result = urlparse(self._key)
-            return ":".join([url_result.hostname, url_result.path])
-        else:
-            return super(SitePath, self).segment
+    def __init__(self, host_name, relative_path, parent=None):
+        super(SitePath, self).__init__(
+            ":".join([host_name, relative_path]), parent, ResourcePath("sites")
+        )
 
-    @property
-    def collection(self):
-        if self._collection is None:
-            self._collection = ResourcePath("sites")
-        return self._collection
+    @staticmethod
+    def from_url(url, parent=None):
+        url_result = urlparse(url)
+        return SitePath(url_result.hostname, url_result.path, parent)
diff --git a/office365/onedrive/sites/sites_with_root.py b/office365/onedrive/sites/sites_with_root.py
@@ -21,12 +21,24 @@ def get_all_sites(self):
         self.context.add_query(qry)
         return return_type
 
+    def get_by_path(self, path):
+        """Address Site resource by server relative path
+
+        :param str path: Server relative path
+        """
+        tenant_part = self.context.tenant_name.split(".")[0]
+        host_name = "{0}.sharepoint.com".format(tenant_part)
+        return_type = Site(self.context, SitePath(host_name, path, self.resource_path))
+        qry = ReadEntityQuery(return_type)
+        self.context.add_query(qry)
+        return return_type
+
     def get_by_url(self, url):
         """Address Site resource by absolute url
 
         :param str url: Site absolute url
         """
-        return_type = Site(self.context, SitePath(url, self.resource_path))
+        return_type = Site(self.context, SitePath.from_url(url, self.resource_path))
         qry = ReadEntityQuery(return_type)
         self.context.add_query(qry)
         return return_type
diff --git a/office365/teams/teamwork/tags/member.py b/office365/teams/teamwork/tags/member.py
@@ -1,5 +1,16 @@
+from typing import Optional
+
 from office365.entity import Entity
 
 
 class TeamworkTagMember(Entity):
     """Represents a user in a team to whom a tag is applied."""
+
+    def __str__(self):
+        return self.display_name or self.entity_type_name
+
+    @property
+    def display_name(self):
+        # type: () -> Optional[str]
+        """The member's display name."""
+        return self.properties.get("displayName", None)
diff --git a/tests/onedrive/test_content_type.py b/tests/onedrive/test_content_type.py
@@ -45,3 +45,9 @@ def test2_create_site_content_type(self):
     def test5_delete(self):
         ct_to_del = self.__class__.target_ct
         ct_to_del.delete_object().execute_query()
+
+    def test6_get_applicable_content_types_for_list(self):
+        site = self.client.sites.root
+        doc_lib = site.lists["Documents"].get().execute_query()
+        cts = site.get_applicable_content_types_for_list(doc_lib.id).execute_query()
+        self.assertIsNotNone(cts.resource_path)
diff --git a/tests/onedrive/test_site.py b/tests/onedrive/test_site.py
@@ -1,4 +1,6 @@
 from office365.onedrive.sites.site import Site
+from tests import test_team_site_url
+from tests.decorators import requires_delegated_permission
 from tests.graph_case import GraphTestCase
 
 
@@ -18,38 +20,52 @@ def setUpClass(cls):
     def tearDownClass(cls):
         pass
 
+    @requires_delegated_permission("Sites.Read.All", "Sites.ReadWrite.All")
     def test1_get_root_site(self):
-        root_site = self.test_site.get().execute_query()
-        assert root_site.id is not None
+        result = self.client.sites.root.get().execute_query()
+        assert result.id is not None
 
+    @requires_delegated_permission("Sites.Read.All", "Sites.ReadWrite.All")
     def test2_get_site_by_path(self):
-        pass
+        result = self.client.sites.get_by_path("/sites/team").execute_query()
+        self.assertIsNotNone(result.resource_path)
+
+    @requires_delegated_permission("Sites.Read.All", "Sites.ReadWrite.All")
+    def test3_get_site_by_url(self):
+        result = self.client.sites.get_by_url(test_team_site_url).execute_query()
+        self.assertIsNotNone(result.resource_path)
 
-    def test3_get_activities_by_interval(self):
+    @requires_delegated_permission(
+        "Files.Read",
+        "Files.ReadWrite",
+        "Files.Read.All",
+        "Files.ReadWrite.All",
+        "Sites.Read.All",
+        "Sites.ReadWrite.All",
+    )
+    def test4_get_activities_by_interval(self):
         col = self.test_site.get_activities_by_interval().execute_query()
         self.assertIsNotNone(col)
 
-    def test4_list_followed_sites(self):
-        sites = self.client.me.followed_sites.get().execute_query()
-        self.followed_sites_count = len(sites)
-
+    @requires_delegated_permission("Sites.ReadWrite.All")
     def test5_follow(self):
-        pass
+        self.client.me.follow_site(self.test_site).execute_query()
 
-    def test6_unfollow(self):
-        pass
+    @requires_delegated_permission("Sites.Read.All", "Sites.ReadWrite.All")
+    def test6_list_followed_sites(self):
+        sites = self.client.me.followed_sites.get().execute_query()
+        self.followed_sites_count = len(sites)
+        self.assertGreaterEqual(len(sites), 1, "No followed sites were returned")
 
-    def test7_get_applicable_content_types_for_list(self):
-        site = self.client.sites.root
-        doc_lib = site.lists["Documents"].get().execute_query()
-        cts = site.get_applicable_content_types_for_list(doc_lib.id).execute_query()
-        self.assertIsNotNone(cts.resource_path)
+    @requires_delegated_permission("Sites.ReadWrite.All")
+    def test7_unfollow(self):
+        self.client.me.unfollow_site(self.test_site).execute_query()
 
-    def test8_get_operations(self):
+    def test9_get_operations(self):
         ops = self.client.sites.root.operations.get().execute_query()
         self.assertIsNotNone(ops.resource_path)
 
-    def test9_get_analytics(self):
+    def test_10_get_analytics(self):
         site = self.client.sites.root
         result = site.analytics.last_seven_days.get().execute_query()
         self.assertIsNotNone(result.resource_path)
diff --git a/tests/outlook/test_event.py b/tests/outlook/test_event.py
@@ -2,12 +2,14 @@
 
 from office365.outlook.calendar.events.event import Event
 from tests import test_user_principal_name
+from tests.decorators import requires_delegated_permission
 from tests.graph_case import GraphTestCase
 
 
 class TestOutlookEvent(GraphTestCase):
     target_event = None  # type: Event
 
+    @requires_delegated_permission("Calendars.ReadWrite", "Calendars.ReadWrite.Shared")
     def test2_create_event(self):
         when = datetime.now() + timedelta(days=1)
         new_event = self.client.me.calendar.events.add(
diff --git a/tests/reports/test_insights.py b/tests/reports/test_insights.py
@@ -1,8 +1,25 @@
+from tests.decorators import requires_delegated_permission
 from tests.graph_case import GraphTestCase
 
 
 class TestInsights(GraphTestCase):
 
+    @requires_delegated_permission("Sites.Read.All", "Sites.ReadWrite.All")
     def test1_list_trending(self):
         result = self.client.me.insights.trending.get().execute_query()
         self.assertIsNotNone(result.resource_path)
+
+    @requires_delegated_permission("Sites.Read.All", "Sites.ReadWrite.All")
+    def test2_list_shared(self):
+        result = self.client.me.insights.shared.get().execute_query()
+        self.assertIsNotNone(result.resource_path)
+
+    @requires_delegated_permission("Sites.Read.All", "Sites.ReadWrite.All")
+    def test3_list_used(self):
+        result = self.client.me.insights.used.get().execute_query()
+        self.assertIsNotNone(result.resource_path)
+
+    @requires_delegated_permission("User.Read", "User.Read.All", "User.ReadWrite.All")
+    def test4_get_user_settings(self):
+        result = self.client.me.settings.get().execute_query()
+        self.assertIsNotNone(result.resource_path)
diff --git a/tests/reports/test_security_reports.py b/tests/reports/test_security_reports.py
@@ -1,3 +1,4 @@
+from tests.decorators import requires_delegated_permission
 from tests.graph_case import GraphTestCase
 
 
@@ -10,8 +11,15 @@ def setUpClass(cls):
     def tearDownClass(cls):
         pass
 
+    # @requires_delegated_permission("AttackSimulation.Read.All")
     def test1_get_attack_simulation_repeat_offenders(self):
         result = (
             self.client.reports.security.get_attack_simulation_repeat_offenders().execute_query()
         )
         self.assertIsNotNone(result.value)
+
+    def test2_get_attack_simulation_simulation_user_coverage(self):
+        result = (
+            self.client.reports.security.get_attack_simulation_simulation_user_coverage().execute_query()
+        )
+        self.assertIsNotNone(result.value)
