setup: include ujson as required dependency

also update psutil version constraint

Signed-off-by: Rui Chen <rui@chenrui.dev>

1 files changed, 10 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 9344d0b0..a818ade1 100755
--- a/setup.py
+++ b/setup.py
@@ -41,7 +41,16 @@ def get_data_files():
 
 
 def get_install_requires():
-    requires = ['psutil>=5.3.0', 'defusedxml', 'future', 'packaging']
+    requires = [
+        'psutil>=5.6.7',
+        'defusedxml',
+        'packaging',
+        'future; python_version < "3.0"',
+        'ujson<3; python_version < "3.0"',
+        'ujson<4; python_version >= "3.5" and python_version < "3.6"',
+        'ujson<5; python_version >= "3.6" and python_version < "3.7"',
+        'ujson>=5.4.0; python_version >= "3.7"',
+    ]
     if sys.platform.startswith('win'):
         requires.append('bottle')
         requires.append('requests')