diff --git a/mwclient/listing.py b/mwclient/listing.py index 3c9edb8f5724c8b058f1af480c810d632d3881a2..0c6514881c086683d7097adb90c3b9b96978870e 100644 --- a/mwclient/listing.py +++ b/mwclient/listing.py @@ -128,17 +128,11 @@ class List(object): @staticmethod def get_prefix(prefix, generator=False): - if generator: - return 'g' + prefix - else: - return prefix + return ('g' if generator else '') + prefix @staticmethod def get_list(generator=False): - if generator: - return GeneratorList - else: - return List + return GeneratorList if generator else List class NestedList(List):