possibly new spicemanip

sopel_modules/spicemanip/__init__.py

@@ -16,37 +16,292 @@ class Spicemanip():
     def __init__(self):
         pass
 
-    def __call__(self, inputs, task=None, output_type=None):
+    def __call__(self, inputs, task, output_type='default'):
+
+        if hasattr(self, task):
+            raise Exception(str(self.task) + " is not a valid spicemanip function.")
+
+        # convert input into list
         self.inputs = inputs
-        self.inputs_list = self.input_handler(self.inputs)
+        self.core_input_handler()
+
+        # determine task
+        self.task = task
+        self.core_task_handler()
+
+        # determine output of list
         self.output_type = output_type
-        return self.core()
+        self.core_output_handler()
 
-    def input_handler(self, inputs):
+        # cleanup
+        self.core_class_cleanup()
+
+        # return value/s
+        return self.outputs
+
+    def core_class_cleanup(self):
+        for classitem in ["inputs", "task", "output_type"]:
+            if hasattr(self, classitem):
+                del classitem
+
+    def core_input_handler(self):
         # Input needs to be a list, but don't split a word into letters
-        if not inputs:
-            input_list = []
-        if isinstance(inputs, list):
-            input_list = inputs
-        if isinstance(inputs, collections.abc.KeysView):
-            input_list = list(inputs)
-        if isinstance(inputs, dict):
-            input_list = list(inputs.keys())
+        if not self.inputs:
+            self.input_list = []
+        if isinstance(self.inputs, list):
+            self.input_list = list(self.inputs)
+        if isinstance(self.inputs, collections.abc.KeysView):
+            self.input_list = list(self.inputs)
+        if isinstance(self.inputs, dict):
+            self.input_list = list(self.inputs.keys())
+        if not isinstance(self.inputs, list):
+            self.input_list = list(self.inputs.split(" "))
+
+    def core_task_handler(self):
+
+        # Create return
+        if self.task in ['create']:
+            self.outputs = self.input_list
+            return
+
+        """
+        returnvalue = eval('spicemanip_' + outputtask + '(inputs, outputtask, mainoutputtask, suboutputtask)')
+        """
+
+        if len(self.input_list):
+
+            if self.task in [0, 'complete']:
+                self.outputs = self.string(self.input_list)
+                return
+
+            elif self.task in ['index']:
+                self.outputs = self.index(self.input_list[0], self.input_list[1], self.input_list[2])
+
+            elif self.is_digit(self.task):
+                self.outputs = self.number(self.input_list, int(self.task))
+
+            elif self.task in ['last']:
+                self.outputs = self.number(self.input_list, -1)
+
+            elif "^" in str(self.task):
+                range_start, range_end = str(self.task).split("^", 1).sort()
+                self.outputs = self.rangebetween(self.input_list, range_start, range_end)
+
+            elif str(self.task).startswith("split_"):
+                self.outputs = self.split(self.input_list, str(self.task).replace("split_", ""))
+
+            elif str(self.task).endswith("!"):
+                self.outputs = self.exclude(self.input_list, str(self.task).replace("!", ""))
+
+            elif str(self.task).endswith("+"):
+                self.outputs = self.incrange_plus(self.input_list, str(self.task).replace("+", ""))
+
+            elif str(self.task).endswith("-"):
+                self.outputs = self.incrange_minus(self.input_list, str(self.task).replace("-", ""))
+
+            elif str(self.task).endswith(">"):
+                self.outputs = self.excrange_plus(self.input_list, str(self.task).replace(">", ""))
+
+            elif str(self.task).endswith("<"):
+                self.outputs = self.excrange_minus(self.input_list, str(self.task).replace("<", ""))
+
+            elif hasattr(self, self.task) and not str(self.task).startswith("core_"):
+                self.outputs = eval('self.' + str(self.task) + '(inputs)')
+
+            else:
+                raise Exception(str(self.task) + " is not a valid spicemanip function.")
+
+    def core_output_handler(self):
+        if not self.output_type:
+            self.output_type = "list"
+        self.outputs = self.inputs_list
+
+        # default return if not specified
+        if self.output_type == 'default':
+            if self.task in [
+                                'string', 'number', 'rangebetween', 'exclude', 'random',
+                                'incrange_plus', 'incrange_minus', 'excrange_plus', 'excrange_minus'
+                                ]:
+                self.output_type = 'string'
+            elif self.task in ['count']:
+                self.output_type = 'dict'
+
+        # verify output is correct
+        if self.output_type == 'return':
+            return self.outputs
+        if self.output_type == 'string':
+            self.outputs = self.string(self.outputs)
+        elif self.output_type in ['list', 'array']:
+            self.outputs = self.list(self.outputs)
+
+    # Convert list to string
+    def string(self, inputs):
         if not isinstance(inputs, list):
-            input_list = list(inputs.split(" "))
-            input_list = [x for x in input_list if x and x not in ['', ' ']]
-            input_list = [inputspart.strip() for inputspart in input_list]
-        return input_list
+            return inputs
+        if not inputs or not len(inputs):
+            return ''
+        return ' '.join(inputs)
 
-    def output_handler(self, inputs_list, output_type):
-        if not output_type:
-            output_type = "list"
-        outputs = inputs_list
-        return outputs
+    # Convert inputs to list
+    def list(self, inputs):
+        if isinstance(inputs, list):
+            return inputs
+        if not inputs or not len(inputs):
+            return []
+        return list(inputs.split(" "))
 
-    def core(self):
-        outputs = self.output_handler(self.inputs_list, self.output_type)
-        return outputs
+    def index(self, indexitem, listtoindex, listtocompare):
+        for x, y in zip(listtoindex, listtocompare):
+            if x == indexitem:
+                return y
+        return ''
+
+    # Get number item from list
+    def number(self, inputs, index_number):
+        if int(index_number) > 0:
+            index_number -= 1
+        try:
+            return inputs[int(index_number)]
+        except IndexError:
+            return ''
+
+    # range between items in list
+    def rangebetween(self, inputs, range_start, range_end):
+        range_start, range_end = int(range_start), int(range_end)
+        if range_end == range_start:
+            return self.number(inputs, range_start)
+        if range_start < 0:
+            range_start = 1
+        if range_end > len(inputs):
+            range_end = len(inputs)
+        newlist = []
+        for i in range(range_start, range_end + 1):
+            newlist.append(str(self.number(inputs, i)))
+        return ' '.join(newlist)
+
+    # split list by string
+    def split(self, inputs, splitby):
+        split_list = []
+        restring = ' '.join(inputs)
+        if splitby not in inputs:
+            split_list = [restring]
+        else:
+            split_list = restring.split(splitby)
+        return ' '.join(split_list)
+
+    # exclude number
+    def exclude(self, inputs, excludenum):
+        try:
+            del inputs[int(excludenum) - 1]
+            return ' '.join(inputs)
+        except IndexError:
+            return ' '.join(inputs)
+
+    # Forward Range includes index number
+    def incrange_plus(self, inputs, range_start):
+        return self.rangebetween(inputs, int(range_start), len(inputs))
+
+    # Reverse Range includes index number
+    def incrange_minus(self, inputs, range_end):
+        return self.rangebetween(inputs, 1, int(range_end))
+
+    # Forward Range excludes index number
+    def excrange_plus(self, inputs, range_start):
+        return self.rangebetween(inputs, int(range_start) + 1, len(inputs))
+
+    # Reverse Range excludes index number
+    def excrange_minus(self, inputs, range_end):
+        return self.rangebetween(inputs, 1, int(range_end) - 1)
+
+    # dedupe list
+    def dedupe(self, inputs):
+        newlist = []
+        for inputspart in inputs:
+            if inputspart not in newlist:
+                newlist.append(inputspart)
+        return newlist
+
+    # Sort list
+    def sort(self, inputs):
+        return sorted(inputs)
+
+    # reverse sort list
+    def rsort(self, inputs):
+        return sorted(inputs)[::-1]
+
+    # count items in list, return dictionary
+    def count(self, inputs):
+        returndict = dict()
+        uniqueinputitems, uniquecount = [], []
+        for inputspart in inputs:
+            if inputspart not in uniqueinputitems:
+                uniqueinputitems.append(inputspart)
+        for uniqueinputspart in uniqueinputitems:
+            count = 0
+            for ele in inputs:
+                if (ele == uniqueinputspart):
+                    count += 1
+            uniquecount.append(count)
+        for inputsitem, unumber in zip(uniqueinputitems, uniquecount):
+            returndict[inputsitem] = unumber
+        return returndict
+
+    # random item from list
+    def random(self, inputs):
+        randomselectlist = list(inputs)
+        while len(randomselectlist) > 1:
+            random.shuffle(randomselectlist)
+            randomselect = randomselectlist[random.randint(0, len(randomselectlist) - 1)]
+            randomselectlist.remove(randomselect)
+        randomselect = randomselectlist[0]
+        return randomselect
+
+    # remove random item from list
+    def exrandom(self, inputs):
+        randremove = self.random(inputs)
+        inputs.remove(randremove)
+        return inputs
+
+    # Convert list into lowercase
+    def lower(self, inputs):
+        return [inputspart.lower() for inputspart in inputs]
+
+    # Convert list to uppercase
+    def upper(self, inputs):
+        return [inputspart.upper() for inputspart in inputs]
+
+    # Convert list to uppercase
+    def title(self, inputs):
+        return [inputspart.title() for inputspart in inputs]
+
+    # Reverse List Order
+    def reverse(self, inputs):
+        return inputs[::-1]
+
+    def list_nospace(self, inputs):
+        return ','.join(str(x) for x in inputs)
+
+    # comma seperated list with and
+    def andlist(self, inputs):
+        if len(inputs) < 2:
+            return ' '.join(inputs)
+        lastentry = str("and " + str(inputs[len(inputs) - 1]))
+        del inputs[-1]
+        inputs.append(lastentry)
+        if len(inputs) == 2:
+            return ' '.join(inputs)
+        return ', '.join(str(x) for x in inputs)
+
+    # comma seperated list with or
+    def orlist(self, inputs):
+        if len(inputs) < 2:
+            return ' '.join(inputs)
+        lastentry = str("or " + str(inputs[len(inputs) - 1]))
+        del inputs[-1]
+        inputs.append(lastentry)
+        if len(inputs) == 2:
+            return ' '.join(inputs)
+        return ', '.join(str(x) for x in inputs)
 
 
 core = Spicemanip()
@@ -193,7 +448,7 @@ def spicemanip_rsort(inputs, outputtask, mainoutputtask, suboutputtask):
 # count items in list, return dictionary
 def spicemanip_count(inputs, outputtask, mainoutputtask, suboutputtask):
     returndict = dict()
-    if inputs == []:
+    if not len(inputs):
         return returndict
     uniqueinputitems, uniquecount = [], []
     for inputspart in inputs:
@@ -212,7 +467,7 @@ def spicemanip_count(inputs, outputtask, mainoutputtask, suboutputtask):
 
 # random item from list
 def spicemanip_random(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     randomselectlist = []
     for temppart in inputs:
@@ -227,7 +482,7 @@ def spicemanip_random(inputs, outputtask, mainoutputtask, suboutputtask):
 
 # remove random item from list
 def spicemanip_exrandom(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return []
     randremove = spicemanip_random(inputs, outputtask, mainoutputtask, suboutputtask)
     inputs.remove(randremove)
@@ -236,48 +491,48 @@ def spicemanip_exrandom(inputs, outputtask, mainoutputtask, suboutputtask):
 
 # Convert list into lowercase
 def spicemanip_lower(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return [inputspart.lower() for inputspart in inputs]
 
 
 # Convert list to uppercase
 def spicemanip_upper(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return [inputspart.upper() for inputspart in inputs]
 
 
 # Convert list to uppercase
 def spicemanip_title(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return [inputspart.title() for inputspart in inputs]
 
 
 # Reverse List Order
 def spicemanip_reverse(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return []
     return inputs[::-1]
 
 
 # comma seperated list
 def spicemanip_list(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return ', '.join(str(x) for x in inputs)
 
 
 def spicemanip_list_nospace(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return ','.join(str(x) for x in inputs)
 
 
 # comma seperated list with and
 def spicemanip_andlist(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     if len(inputs) < 2:
         return ' '.join(inputs)
@@ -291,7 +546,7 @@ def spicemanip_andlist(inputs, outputtask, mainoutputtask, suboutputtask):
 
 # comma seperated list with or
 def spicemanip_orlist(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     if len(inputs) < 2:
         return ' '.join(inputs)
@@ -305,7 +560,7 @@ def spicemanip_orlist(inputs, outputtask, mainoutputtask, suboutputtask):
 
 # exclude number
 def spicemanip_exclude(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     del inputs[int(mainoutputtask) - 1]
     return ' '.join(inputs)
@@ -313,14 +568,14 @@ def spicemanip_exclude(inputs, outputtask, mainoutputtask, suboutputtask):
 
 # Convert list to string
 def spicemanip_string(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return ' '.join(inputs)
 
 
 # Get number item from list
 def spicemanip_number(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     elif len(inputs) == 1:
         return inputs[0]
@@ -332,14 +587,14 @@ def spicemanip_number(inputs, outputtask, mainoutputtask, suboutputtask):
 
 # Get Last item from list
 def spicemanip_last(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return inputs[len(inputs) - 1]
 
 
 # range between items in list
 def spicemanip_rangebetween(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     if not str(mainoutputtask).isdigit() or not str(suboutputtask).isdigit():
         return ''
@@ -362,27 +617,27 @@ def spicemanip_rangebetween(inputs, outputtask, mainoutputtask, suboutputtask):
 
 # Forward Range includes index number
 def spicemanip_incrange_plus(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return spicemanip_rangebetween(inputs, outputtask, int(mainoutputtask), len(inputs))
 
 
 # Reverse Range includes index number
 def spicemanip_incrange_minus(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return spicemanip_rangebetween(inputs, outputtask, 1, int(mainoutputtask))
 
 
 # Forward Range excludes index number
 def spicemanip_excrange_plus(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return spicemanip_rangebetween(inputs, outputtask, int(mainoutputtask) + 1, len(inputs))
 
 
 # Reverse Range excludes index number
 def spicemanip_excrange_minus(inputs, outputtask, mainoutputtask, suboutputtask):
-    if inputs == []:
+    if not len(inputs):
         return ''
     return spicemanip_rangebetween(inputs, outputtask, 1, int(mainoutputtask) - 1)