Skip to content

Commit

Permalink
Merge branch 'dev' into ntindle/open-2101-integration-test-submit-an-…
Browse files Browse the repository at this point in the history
…agent-to-the-marketplace-dev
  • Loading branch information
ntindle authored Jan 26, 2025
2 parents 1c25b32 + 479a847 commit ee2a2ed
Showing 1 changed file with 0 additions and 130 deletions.
130 changes: 0 additions & 130 deletions autogpt_platform/backend/backend/blocks/twitter/users/blocks.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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)

0 comments on commit ee2a2ed

Please sign in to comment.