paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-10-23 04:52:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

tools: Fix mypy errors with generate-integration-docs-screenshot.

Browse Source 
3f4d0f72fd adjusted the types in
preparation for extending the functionality, but only in later commits
that are not merged yet did it make these updates to the types.
This commit is contained in:
Tim Abbott
2021-04-28 08:47:38 -07:00
parent 91c81eadd0
commit 2d9631191e
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tools/generate-integration-docs-screenshot
View File

@@ -41,6 +41,7 @@ from zerver.lib.actions import (
from zerver.lib.integrations import (
DOC_SCREENSHOT_CONFIG,
INTEGRATIONS,
BaseScreenshotConfig,
Integration,
ScreenshotConfig,
WebhookIntegration,
@@ -54,9 +55,7 @@ from zerver.lib.webhooks.common import get_fixture_http_headers
from zerver.models import Message, UserProfile, get_realm, get_user_by_delivery_email
def create_integration_bot(
integration: WebhookIntegration, bot_name: Optional[str] = None
) -> UserProfile:
def create_integration_bot(integration: Integration, bot_name: Optional[str] = None) -> UserProfile:
realm = get_realm("zulip")
owner = get_user_by_delivery_email("iago@zulip.com", realm)
bot_email = f"{integration.name}-bot@example.com"
@@ -87,7 +86,7 @@ def create_integration_bot(
return bot
def create_integration_stream(integration: WebhookIntegration, bot: UserProfile) -> None:
def create_integration_stream(integration: Integration, bot: UserProfile) -> None:
assert isinstance(bot.bot_owner, UserProfile)
realm = bot.bot_owner.realm
stream, created = create_stream_if_needed(realm, integration.stream_name)
@@ -204,13 +203,17 @@ def capture_last_message_screenshot(bot: UserProfile, image_path: str) -> None:
def generate_screenshot_from_config(
integration_name: str, screenshot_config: ScreenshotConfig
integration_name: str, screenshot_config: BaseScreenshotConfig
) -> None:
integration = get_integration(integration_name)
fixture_path, image_path = get_fixture_and_image_paths(integration, screenshot_config)
bot = create_integration_bot(integration, screenshot_config.bot_name)
create_integration_stream(integration, bot)
message_sent = send_bot_payload_message(bot, integration, fixture_path, screenshot_config)
if isinstance(integration, WebhookIntegration):
assert isinstance(screenshot_config, ScreenshotConfig)
message_sent = send_bot_payload_message(bot, integration, fixture_path, screenshot_config)
else:
raise AssertionError("Not yet implemented")
if message_sent:
capture_last_message_screenshot(bot, image_path)
print(f"Screenshot captured to: {BOLDRED}{image_path}{ENDC}")