Now code supports freeze into exe

Author: miloserdow

Diff for: .travis.yml

@@ -1,9 +1,9 @@
 language: python
 python:
-  - "3.3"
   - "3.4"
 cache: pip
 install:
   - pip install -r requirements.txt
 script:
+  - python setup.py build
   - nosetests

Diff for: README.md

@@ -5,3 +5,6 @@ The goal of leet however is to be simpler and as user friendly as possible, maki
 For a more complete description of the leet project, it has been decided to link the Sourceforge pages describing the software application as well as the encryption algorithm to be used.
 
 Here: https://sourceforge.net/p/home-of-leet/wiki/browse_pages/
+
+## Build instructions
+Use: python setup.py build

Diff for: driver.py

@@ -5,7 +5,14 @@
 import os
 import sys
 
-sys.path.append(os.path.join(os.path.dirname(__file__), "src"))
+if getattr(sys, 'frozen', False):
+    # frozen
+    dir_ = os.path.dirname(sys.executable)
+else:
+    # unfrozen
+    dir_ = os.path.dirname(os.path.realpath(__file__))
+
+sys.path.append(os.path.join(dir_, "src"))
 
 import curses
 import logging

Diff for: requirements.txt

@@ -1,2 +1,3 @@
-pycrypto >= 2.6
-psutil >= 5.4.3
+cx_freeze   == 5.0.1
+pycrypto    >= 2.6
+psutil      >= 5.4.3

Diff for: setup.py

@@ -1,11 +1,16 @@
 #!/usr/bin/env python
 
-from setuptools import setup, find_packages
+import sys, os
+from cx_Freeze import setup, Executable
+
+dir_ = os.path.dirname(os.path.realpath(__file__))
+sys.path.append(os.path.join(dir_, "src"))
 
 setup(
     name='leet',
     version=1.0,
     author='CCEx',
     description='Encryption tool',
-    license='GPLv3'
+    license='GPLv3',
+    executables = [Executable("driver.py")]
 )

Diff for: src/__init__.py

@@ -63,7 +63,7 @@ def unpad(self, s):
         return s[:-ord(s[len(s) - 1:])]
 
     def encrypt(self, data: str):
-        if not isinstance(self.key, str):
+        if not isinstance(self.key, str) and not isinstance(self.key, bytes):
             raise ValueError('Invalid key')
 
         data = self.pad(data)
@@ -74,7 +74,7 @@ def encrypt(self, data: str):
         return base64.b64encode(C).decode('utf-8')
 
     def decrypt(self, data: str) -> str:
-        if not isinstance(self.key, str):
+        if not isinstance(self.key, str) and not isinstance(self.key, bytes):
             raise ValueError('Invalid key')
 
         data = base64.b64decode(data)