diff --git a/hanzi_chaizi/__init__.py b/hanzi_chaizi/__init__.py index 98da87b..14baed5 100644 --- a/hanzi_chaizi/__init__.py +++ b/hanzi_chaizi/__init__.py @@ -12,7 +12,10 @@ def __init__(self): def query(self, input_char, default=None): result = self.data.get(input_char, default) - return result[0] + if result: + return result[0] + else: + return None if __name__ == "__main__":