Added installation for pip

entro_py_min/__main__.py

@@ -0,0 +1,45 @@
+from . import entro_py_min
+import argparse
+
+
+# List of the arguments one can use to influence the behavior of the program
+parser = argparse.ArgumentParser('entro_py_min', description='Calculate the information entropy of alphabets.')
+
+# INPUT ARGUMENTS
+parser.add_argument('strings', nargs='*', default='', type=str, help='Strings to calculate the entropy of.')
+parser.add_argument('--files', nargs='*', type=str, default='', help='Provide file path(s) to calculate the entropy of.')
+
+# OUTPUT OPTIONS
+parser.add_argument('--simple', nargs='?', type=bool, default=False, help='Determines the explicitness of the output. (True = only entropy shown)')
+parser.add_argument('--max', nargs='?', type=bool, default=False, help='Includes the maximum entropy.')
+
+# CONVERT OPTIONS
+parser.add_argument('--lower', nargs='?', type=bool, default=False, help='Converts given strings or textfiles to lowercase before calculating.')
+parser.add_argument('--upper', nargs='?', type=bool, default=False, help='Converts given strings or textfiles to uppercase before calculating.')
+parser.add_argument('--squash', nargs='?', type=bool, default=False, help='Removes all whitespaces before calculating.')
+args = parser.parse_args()
+
+# Prepares the queue of different strings
+queue = []
+
+# Add all the provided strings to the list
+for string in args.strings:
+    queue.append(string)
+
+# Add all the provided files to the list
+for file in args.files:
+    string = entro_py_min.readEntropyFile(file)
+    queue.append(string)
+
+# Interates over the collected strings and prints the entropies
+for string in queue:
+    if args.lower:
+        string = string.lower()
+    elif args.upper:
+        string = string.upper()
+
+    if args.squash:
+        string = string.replace(" ", "")
+
+    a, b, c = entro_py_min.calculateEntropy(string)
+    entro_py_min.printEntropy(string, a, b, args.simple, (False if not args.max else c))

entro_py_min/entro_py_min.py

@@ -0,0 +1,59 @@
+import math
+
+
+# Calculates the entropy of a given string
+# Returns the entropy and an alphabet with the calculated probabilities
+def calculateEntropy(input_string):
+    alphabet, alphabet_size, entropy = {}, 0, 0
+
+    for char in input_string:
+        if char in alphabet:
+            alphabet[char] += 1
+        else:
+            alphabet[char] = 1
+        alphabet_size += 1
+
+    for char in alphabet:
+        alphabet[char] = alphabet[char] / alphabet_size
+        entropy -= alphabet[char] * math.log(alphabet[char], 2)
+
+    return entropy, alphabet
+
+
+# Calculates the entropy of a given string
+# Returns only the entropy in bits as this is the minimal function
+def calculateEntropyMin(input_string):
+    alphabet, alphabet_size, entropy = {}, 0, 0
+
+    for char in input_string:
+        if char in alphabet:
+            alphabet[char] += 1
+        else:
+            alphabet[char] = 1
+        alphabet_size += 1
+
+    for char in alphabet:
+        i = alphabet[char] / alphabet_size
+        entropy -= i * math.log(i, 2)
+
+    return entropy
+
+
+# Outputs a given entropy including the original text and the alphabet with probabilities
+def printEntropy(original_string, entropy_value, alphabet_dict, simple_bool, max_value):
+    print('---')
+    if not simple_bool:
+        print('Content: ' + original_string)
+        print('Probabilities: ' + str(alphabet_dict))
+    print('Entropy: ' + str(entropy_value) + ' bits')
+    if max_value:
+        print('Maximum Entropy: ' + str(max_value) + ' bits')
+    print('---')
+
+
+# Reads a file by a given path
+def readEntropyFile(path_string):
+    f = open(path_string, 'r')
+    content = f.read().replace('\n', ' ')
+    f.close()
+    return content.strip()