added encrypt decrypt + test and dependency

Author: nevoalm

requirements.txt

@@ -1 +1,2 @@
-requests~=2.22.0
+requests~=2.22.0
+pycrypto~=2.6.1

securenative/sn_crypto.py

@@ -0,0 +1,21 @@
+from Crypto.Cipher import AES
+from Crypto import Random
+from binascii import unhexlify, hexlify
+
+BLOCK_SIZE = 16
+pad = lambda s: s + (BLOCK_SIZE - len(s) % BLOCK_SIZE) * chr(BLOCK_SIZE - len(s) % BLOCK_SIZE)
+
+
+def encrypt(text, cipherKey):
+    iv = Random.new().read(AES.block_size);
+    cipher = AES.new(cipherKey, AES.MODE_CBC, iv)
+    raw = pad(text)
+    return hexlify(iv + cipher.encrypt(raw)).decode('utf-8').strip()
+
+
+def decrypt(encrypted, cipherKey):
+    content = unhexlify(encrypted)
+    iv = content[:BLOCK_SIZE]
+    cipherText = content[BLOCK_SIZE:]
+    aes = AES.new(cipherKey, AES.MODE_CBC, iv)
+    return aes.decrypt(cipherText).decode('utf-8').strip()

securenative/sn_crypto_test.py

@@ -0,0 +1,12 @@
+import unittest
+
+from securenative.sn_crypto import encrypt,decrypt
+
+
+class CryptoTests(unittest.TestCase):
+    def test_encrypt_decrypt(self):
+        api_key = '6EA4915349C0AAC6F6572DA4F6B00C42'
+        data = '{"cid":"198a41ff-a10f-4cda-a2f3-a9ca80c0703b","fp":"6d8cabd95987f8318b1fe01593d5c2a5.24700f9f1986800ab4fcc880530dd0ed"}'
+        self.assertEqual(data,decrypt(encrypt(data,api_key),api_key))
+
+

setup.py

@@ -20,6 +20,7 @@
     long_description_content_type="text/markdown",
     install_requires=[
         "requests",
+        "pycrypro",
     ],
     classifiers=[
         'Development Status :: 5 - Production/Stable',