diff --git a/autogpt_platform/backend/backend/blocks/twitter/users/blocks.py b/autogpt_platform/backend/backend/blocks/twitter/users/blocks.py index ebe45038e81c..62575976ee16 100644 --- a/autogpt_platform/backend/backend/blocks/twitter/users/blocks.py +++ b/autogpt_platform/backend/backend/blocks/twitter/users/blocks.py @@ -23,71 +23,6 @@ from backend.data.model import SchemaField -class TwitterUnblockUserBlock(Block): - """ - Unblock a specific user on Twitter. The request succeeds with no action when the user sends a request to a user they're not blocking or have already unblocked. - """ - - class Input(BlockSchema): - credentials: TwitterCredentialsInput = TwitterCredentialsField( - ["block.write", "users.read", "offline.access"] - ) - - target_user_id: str = SchemaField( - description="The user ID of the user that you would like to unblock", - placeholder="Enter target user ID", - ) - - class Output(BlockSchema): - success: bool = SchemaField(description="Whether the unblock was successful") - error: str = SchemaField(description="Error message if the request failed") - - def __init__(self): - super().__init__( - id="0f1b6570-a631-11ef-a3ea-230cbe9650dd", - description="This block unblocks a specific user on Twitter.", - categories={BlockCategory.SOCIAL}, - input_schema=TwitterUnblockUserBlock.Input, - output_schema=TwitterUnblockUserBlock.Output, - test_input={ - "target_user_id": "12345", - "credentials": TEST_CREDENTIALS_INPUT, - }, - test_credentials=TEST_CREDENTIALS, - test_output=[ - ("success", True), - ], - test_mock={"unblock_user": lambda *args, **kwargs: True}, - ) - - @staticmethod - def unblock_user(credentials: TwitterCredentials, target_user_id: str): - try: - client = tweepy.Client( - bearer_token=credentials.access_token.get_secret_value() - ) - - client.unblock(target_user_id=target_user_id, user_auth=False) - - return True - - except tweepy.TweepyException: - raise - - def run( - self, - input_data: Input, - *, - credentials: TwitterCredentials, - **kwargs, - ) -> BlockOutput: - try: - success = self.unblock_user(credentials, input_data.target_user_id) - yield "success", success - except Exception as e: - yield "error", handle_tweepy_exception(e) - - class TwitterGetBlockedUsersBlock(Block): """ Get a list of users who are blocked by the authenticating user @@ -238,68 +173,3 @@ def run( yield "next_token", next_token except Exception as e: yield "error", handle_tweepy_exception(e) - - -class TwitterBlockUserBlock(Block): - """ - Block a specific user on Twitter - """ - - class Input(BlockSchema): - credentials: TwitterCredentialsInput = TwitterCredentialsField( - ["block.write", "users.read", "offline.access"] - ) - - target_user_id: str = SchemaField( - description="The user ID of the user that you would like to block", - placeholder="Enter target user ID", - ) - - class Output(BlockSchema): - success: bool = SchemaField(description="Whether the block was successful") - error: str = SchemaField(description="Error message if the request failed") - - def __init__(self): - super().__init__( - id="fc258b94-a630-11ef-abc3-df050b75b816", - description="This block blocks a specific user on Twitter.", - categories={BlockCategory.SOCIAL}, - input_schema=TwitterBlockUserBlock.Input, - output_schema=TwitterBlockUserBlock.Output, - test_input={ - "target_user_id": "12345", - "credentials": TEST_CREDENTIALS_INPUT, - }, - test_credentials=TEST_CREDENTIALS, - test_output=[ - ("success", True), - ], - test_mock={"block_user": lambda *args, **kwargs: True}, - ) - - @staticmethod - def block_user(credentials: TwitterCredentials, target_user_id: str): - try: - client = tweepy.Client( - bearer_token=credentials.access_token.get_secret_value() - ) - - client.block(target_user_id=target_user_id, user_auth=False) - - return True - - except tweepy.TweepyException: - raise - - def run( - self, - input_data: Input, - *, - credentials: TwitterCredentials, - **kwargs, - ) -> BlockOutput: - try: - success = self.block_user(credentials, input_data.target_user_id) - yield "success", success - except Exception as e: - yield "error", handle_tweepy_exception(e)