mirrors/legendary
1
0
Fork 0
mirror of https://github.com/derrod/legendary.git synced 2024-12-22 17:55:27 +00:00
Code Issues Projects Releases Wiki Activity

[utils] Improve alias generation and fix explanation

Browse source
This commit is contained in:
derrod 2021-10-02 21:34:12 +02:00
parent e7ce2e5cb7
commit 3f04de448b
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
legendary/utils/aliasing.py
View file

@ -4,7 +4,7 @@ from string import ascii_lowercase, digits
# - name lowercase (without TM etc.) # - name lowercase (without TM etc.)
# - same, but without spaces # - same, but without spaces
# - same, but roman numerals are replaced # - same, but roman numerals are replaced
# if name has >= 3 parts: # if name has >= 2 parts:
# - initials # - initials
# - initials, but roman numerals are intact # - initials, but roman numerals are intact
# - initials, but roman numerals are replaced with number # - initials, but roman numerals are replaced with number
@ -12,10 +12,11 @@ from string import ascii_lowercase, digits
# - run previous recursively with everything before ":" # - run previous recursively with everything before ":"
# if single 'f' in long word: # if single 'f' in long word:
# - split word (this is mainly for cases like Battlfront -> BF) # - split word (this is mainly for cases like Battlfront -> BF)
# the first word longer than 1 character that isn't "the", "for", or "of" will also be added
allowed_characters = ascii_lowercase+digits allowed_characters = ascii_lowercase+digits
roman = { roman = {
# 'i': '1', 'i': '1',
'ii': '2', 'ii': '2',
'iii': '3', 'iii': '3',
'iv': '4', 'iv': '4',
@ -55,7 +56,7 @@ def generate_aliases(game_name, game_folder=None, split_words=True):
] ]
# single word abbreviation # single word abbreviation
first_word = next(i for i in game_parts if i not in ('for', 'the')) first_word = next(i for i in game_parts if i not in ('for', 'the', 'of'))
if len(first_word) > 1: if len(first_word) > 1:
_aliases.append(first_word) _aliases.append(first_word)
# remove subtitle from game # remove subtitle from game
@ -65,7 +66,7 @@ def generate_aliases(game_name, game_folder=None, split_words=True):
if game_folder: if game_folder:
_aliases.extend(generate_aliases(game_folder, split_words=False)) _aliases.extend(generate_aliases(game_folder, split_words=False))
# include initialisms # include initialisms
if len(game_parts) > 2: if len(game_parts) > 1:
_aliases.append(''.join(p[0] for p in game_parts)) _aliases.append(''.join(p[0] for p in game_parts))
_aliases.append(''.join(p[0] if p not in roman else p for p in game_parts)) _aliases.append(''.join(p[0] if p not in roman else p for p in game_parts))
_aliases.append(''.join(roman.get(p, p[0]) for p in game_parts)) _aliases.append(''.join(roman.get(p, p[0]) for p in game_parts))
@ -81,9 +82,10 @@ def generate_aliases(game_name, game_folder=None, split_words=True):
else: else:
new_game_parts.append(word) new_game_parts.append(word)
_aliases.append(''.join(p[0] for p in new_game_parts)) if len(new_game_parts) > 1:
_aliases.append(''.join(p[0] if p not in roman else p for p in new_game_parts)) _aliases.append(''.join(p[0] for p in new_game_parts))
_aliases.append(''.join(roman.get(p, p[0]) for p in new_game_parts)) _aliases.append(''.join(p[0] if p not in roman else p for p in new_game_parts))
_aliases.append(''.join(roman.get(p, p[0]) for p in new_game_parts))
# return sorted unqiues # return sorted unqiues
return sorted(set(_aliases)) return sorted(set(_aliases))