Skip to content

Commit

Permalink
refactor: ruff format
Browse files Browse the repository at this point in the history
  • Loading branch information
maxhoheiser committed Jan 8, 2025
1 parent df859de commit c3cf8ae
Showing 1 changed file with 13 additions and 7 deletions.
20 changes: 13 additions & 7 deletions moto/scheduler/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,9 +126,7 @@ def update(
class ScheduleGroup(BaseModel):
def __init__(self, region: str, account_id: str, name: str):
self.name = name
self.arn = (
f"arn:{get_partition(region)}:scheduler:{region}:{account_id}:schedule-group/{name}"
)
self.arn = f"arn:{get_partition(region)}:scheduler:{region}:{account_id}:schedule-group/{name}"
self.schedules: Dict[str, Schedule] = dict()
self.created_on = None if self.name == "default" else unix_time()
self.last_modified = None if self.name == "default" else unix_time()
Expand Down Expand Up @@ -163,7 +161,9 @@ def __init__(self, region_name: str, account_id: str):
super().__init__(region_name, account_id)
self.schedules: List[Schedule] = list()
self.schedule_groups = {
"default": ScheduleGroup(region=region_name, account_id=account_id, name="default")
"default": ScheduleGroup(
region=region_name, account_id=account_id, name="default"
)
}
self.tagger = TaggingService()

Expand Down Expand Up @@ -262,11 +262,15 @@ def list_schedules(
results.append(schedule)
return results

def create_schedule_group(self, name: str, tags: List[Dict[str, str]]) -> ScheduleGroup:
def create_schedule_group(
self, name: str, tags: List[Dict[str, str]]
) -> ScheduleGroup:
"""
The ClientToken parameter is not yet implemented
"""
group = ScheduleGroup(region=self.region_name, account_id=self.account_id, name=name)
group = ScheduleGroup(
region=self.region_name, account_id=self.account_id, name=name
)
self.schedule_groups[name] = group
self.tagger.tag_resource(group.arn, tags)
return group
Expand All @@ -285,7 +289,9 @@ def list_schedule_groups(self) -> Iterable[ScheduleGroup]:
def delete_schedule_group(self, name: Optional[str]) -> None:
self.schedule_groups.pop(name or "default")

def list_tags_for_resource(self, resource_arn: str) -> Dict[str, List[Dict[str, str]]]:
def list_tags_for_resource(
self, resource_arn: str
) -> Dict[str, List[Dict[str, str]]]:
return self.tagger.list_tags_for_resource(resource_arn)

def tag_resource(self, resource_arn: str, tags: List[Dict[str, str]]) -> None:
Expand Down

0 comments on commit c3cf8ae

Please sign in to comment.