Merge pull request #281 from catgoat/catgoat/domain_blocks

Add admin domain blocks

2 files changed, 105 insertions, 0 deletions

diff --git a/docs/index.rst b/docs/index.rst
index 10e18c0..bb6e5b1 100644
--- a/docs/index.rst
+++ b/docs/index.rst
@@ -882,6 +882,25 @@ Admin account dicts
         'account': # The user's account, as a standard user dict
     }
 
+Admin domain block dicts
+~~~~~~~~~~~~~~~~~~~~~~~~
+.. _domain dicts
+
+.. code-block::python 
+    mastodon.domain_blocks(id=1)
+    #Returns the following dictionary:
+    {
+        'id': #Str. The database id of a domain block,
+        'domain': #Str. The root domain of a block, ie: "example.com",
+        'created_at': #Datetime of the block creation.
+        'severity': #Str. Severity of the domain block, ie: "suspend".
+        'reject_media': #Boolean. True if media is not downloaded from this domain.
+        'reject_reports': #Boolean. True if reports are automatically ignored from this domain.
+        'private_comment': #Str. Private admin comment for a block. None if not set.
+        'public_comment': #Str. Publicly viewable (depending on settings) comment about this domain. None if not set.
+        'obfuscate': #Boolean. True if domain name is obfuscated when listing.
+    }
+
 Status edit dicts
 ~~~~~~~~~~~~~~~~~
 .. _status edit dict:
@@ -1446,6 +1465,10 @@ have admin: scopes attached with a lot of care, but be extra careful with those
 .. automethod:: Mastodon.admin_trending_tags
 .. automethod:: Mastodon.admin_trending_statuses
 .. automethod:: Mastodon.admin_trending_links
+.. automethod:: Mastodon.admin_domain_blocks
+.. automethod:: Mastodon.admin_domain_block
+.. automethod:: Mastodon.admin_update_domain_block
+.. automethod:: Mastodon.admin_delete_domain_block
 
 Acknowledgements
 ----------------
diff --git a/mastodon/Mastodon.py b/mastodon/Mastodon.py
index a60994e..c519a0d 100644
--- a/mastodon/Mastodon.py
+++ b/mastodon/Mastodon.py
@@ -3363,6 +3363,88 @@ class Mastodon:
         params = self.__generate_params(locals())
         return self.__api_request('GET', '/api/v1/admin/trends/links', params)
 
+    @api_version("4.0.0","4.0.0","4.0.0")
+    def admin_domain_blocks(self, id:str=None, limit:int=None):
+        """
+        Fetches a list of blocked domains. Requires scope `admin:read:domain_blocks`.
+
+        Provide an `id` to fetch a specific domain block based on its database id.
+
+        Returns a list of `domain dicts`_, or 404 if a domain is queried for and not found.
+        """
+        id = self.__unpack_id(id)
+        if id is not None:
+            return self.__api_request('GET', '/api/v1/admin/domain_blocks/{0}'.format(id))
+        else:
+            params = self.__generate_params(locals(),['limit'])
+            return self.__api_request('GET', '/api/v1/admin/domain_blocks/', params)
+    
+    @api_version("4.0.0","4.0.0","4.0.0")
+    def admin_domain_block(self, domain:str, severity:str=None, reject_media:bool=None, reject_reports:bool=None, private_comment:str=None, public_comment:str=None, obfuscate:bool=None):
+        """
+        Perform a moderation action on a domain. 
+
+        Valid severities are:
+            * "silence" - hide all posts from federated timelines and do not show notifications to local users from the remote instance's users unless they are following the remote user. 
+            * "suspend" - deny interactions with this instance going forward. This action is reversible.
+            * "limit" - generally used with reject_media=true to force reject media from an instance without silencing or suspending..
+
+        If no action is specified, the domain is only silenced.
+        `domain` is the domain to block. Note that using the top level domain will also imapct all subdomains. ie, example.com will also impact subdomain.example.com.
+        `reject_media` will not download remote media on to your local instance media storage.
+        `reject_reports` ignores all reports from the remote instance.
+        `private_comment` sets a private admin comment for the domain.
+        `public_comment` sets a publicly available comment for this domain, which will be available to local users and may be available to everyone depending on your settings.
+        `obfuscate` sensors some part of the domain name. Useful if the domain name contains unwanted words like slurs.
+        """
+        if domain is None:
+            raise AttributeError("Must provide a domain to block a domain")
+
+        params = self.__generate_params(locals())
+
+
+        self.__api_request('POST', '/api/v1/admin/domain_blocks/', params)
+
+    @api_version("4.0.0","4.0.0","4.0.0")
+    def admin_update_domain_block(self, id:str, severity:str=None, reject_media:bool=None, reject_reports:bool=None, private_comment:str=None, public_comment:str=None, obfuscate:bool=None):
+        """
+        Modify existing moderation action on a domain. 
+
+        Valid severities are:
+            * "silence" - hide all posts from federated timelines and do not show notifications to local users from the remote instance's users unless they are following the remote user. 
+            * "suspend" - deny interactions with this instance going forward. This action is reversible.
+            * "limit" - generally used with reject_media=true to force reject media from an instance without silencing or suspending.
+
+        If no action is specified, the domain is only silenced.
+        `domain` is the domain to block. Note that using the top level domain will also imapct all subdomains. ie, example.com will also impact subdomain.example.com.
+        `reject_media` will not download remote media on to your local instance media storage.
+        `reject_reports` ignores all reports from the remote instance.
+        `private_comment` sets a private admin comment for the domain.
+        `public_comment` sets a publicly available comment for this domain, which will be available to local users and may be available to everyone depending on your settings.
+        `obfuscate` sensors some part of the domain name. Useful if the domain name contains unwanted words like slurs.
+        """
+        if id is None:
+            raise AttributeError("Must provide an id to modify the existing moderation actions on a given domain.")
+
+        params = self.__generate_params(locals())
+
+        self.__api_request('PUT', '/api/v1/admin/domain_blocks/', params)
+
+    @api_version("4.0.0","4.0.0","4.0.0")
+    def admin_delete_domain_blocks(self, id:str=None):
+        """
+        Removes moderation action against a given domain. Requires scope `admin:write:domain_blocks`.
+
+        Provide an `id` to remove a specific domain block based on its database id.
+
+        Returns 200 OK if successful.
+        """
+        id = self.__unpack_id(id)
+        if id is not None:
+            return self.__api_request('DELETE', '/api/v1/admin/domain_blocks/{0}'.format(id))
+        else:
+            raise AttributeError("You must provide an id of an existing domain block to remove it.")
+
     ###
     # Push subscription crypto utilities
     ###