Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support for Age Rating #73

Merged
merged 2 commits into from
Oct 19, 2022
Merged

Support for Age Rating #73

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ec95096
Update project deps
bpepple Oct 14, 2022
9effec0
Add support for age ratings
bpepple Oct 19, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 16 additions & 10 deletions metrontagger/talker.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -173,18 +173,11 @@ def retrieve_single_issue(self, fn: Path, id: int) -> None:
self._write_issue_md(fn, id)

@staticmethod
def _map_resp_to_metadata(resp: IssueSchema) -> Metadata:
def _map_resp_to_metadata(resp: IssueSchema) -> Metadata: # NOQA C901
# Helper functions
def create_resource_list(resource) -> List[Basic]:
return [Basic(r.name, r.id) for r in resource]

def create_reprint_list(resource) -> List[Basic]:
return [Basic(r.issue, r.id) for r in resource]

def create_stories_list(resource) -> List[Basic]:
# Metron doesn't have a story id field
return [Basic(story) for story in resource]

def create_note(issue_id: int) -> str:
now_date = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
return f"Tagged with MetronTagger-{__version__} using info from Metron on {now_date}. [issue_id:{issue_id}]"
Expand All @@ -202,6 +195,17 @@ def create_role_list(roles: List[RolesSchema]) -> List[Role]:
md.add_credit(Credit(c.creator, [], c.id))
return md

def map_ratings(rating: str) -> str:
age_rating = rating.lower()
if age_rating == "everyone":
return "Everyone"
elif age_rating in {"teen", "teen plus"}:
return "Teen"
elif age_rating == "mature":
return "Mature 17+"
else:
return "Unknown"

md = Metadata()
md.info_source = Basic("Metron", resp.id)
md.series = Series(
Expand All @@ -217,7 +221,7 @@ def create_role_list(roles: List[RolesSchema]) -> List[Role]:
md.comments = resp.desc
md.notes = create_note(md.info_source.id_)
if resp.story_titles:
md.stories = create_stories_list(resp.story_titles)
md.stories = [Basic(story) for story in resp.story_titles]
if resp.characters:
md.characters = create_resource_list(resp.characters)
if resp.teams:
Expand All @@ -227,8 +231,10 @@ def create_role_list(roles: List[RolesSchema]) -> List[Role]:
if resp.series.genres:
md.genres = create_resource_list(resp.series.genres)
if resp.reprints:
md.reprints = create_reprint_list(resp.reprints)
md.reprints = [Basic(r.issue, r.id) for r in resp.reprints]
if resp.credits:
md = add_credits_to_metadata(md, resp.credits)
if resp.rating:
md.age_rating = map_ratings(resp.rating.name)

return md
Loading