cli: use pass_obj instead off pass_context for commands

This commit is contained in:
Corentin Sechet 2022-04-11 12:38:03 +02:00
parent 52c07a934c
commit 9ef40a3a12
1 changed files with 5 additions and 7 deletions

View File

@ -6,7 +6,7 @@ from typing import Any, Optional
from click import Context as ClickContext
from click import Path as PathArgument
from click import argument, group, option, pass_context
from click import argument, group, option, pass_context, pass_obj
from frontools.config import Config
from frontools.css import css_diff
@ -50,11 +50,10 @@ async def main(
@main.command(name="css-diff")
@argument("right_source", type=str)
@pass_context
@pass_obj
@_async_command
async def css_diff_cli(ctx: ClickContext, right_source: str) -> None:
async def css_diff_cli(config: Config, right_source: str) -> None:
"""Diff CSS"""
config: Config = ctx.obj
for _, site in config.sites:
for site_url in site.urls:
await css_diff(
@ -67,15 +66,14 @@ async def css_diff_cli(ctx: ClickContext, right_source: str) -> None:
@option("-o", "--output-directory", type=PathArgument(), default=None)
@option("-r", "--resolution", type=str, default=None)
@_async_command
@pass_context
@pass_obj
async def screenshot_diff_cli(
ctx: ClickContext,
config: Config,
source: str,
output_directory: Optional[str],
resolution: Optional[str],
) -> None:
"""Generate screenshot diffs"""
config: Config = ctx.obj
await screenshot_diff(config, source, output_directory, resolution=resolution)