From 4d4307c7e2d760e8b2fee1b368f5e5939d3b9f5d Mon Sep 17 00:00:00 2001
From: Unnamed <miaoluda@users.noreply.github.com>
Date: Sat, 28 Mar 2020 22:25:49 +0800
Subject: [PATCH] Update __init__.py: fix nonetype error

TypeError: 'NoneType' object is not subscriptable if input_char not in the dict.
---
 hanzi_chaizi/__init__.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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__":