HelperFunctions.py

from tabulate import tabulate


class HelperFunctions:

    #converts numbers to an int
    @staticmethod
    def ConvertToInt(num):
        
        return int(str(num).replace(',', ''))
    

    #Gets the users input for a given question, has to be between the min and max values (inclusive)
    @staticmethod
    def GetQuestionInput(question, min, max):

        option = -1

        try:

            option = int(input(question))

        except ValueError:

            print("Not a valid input was given")

        while not (min <= option <= max):

            try:

                option = int(input(question + " Enter a value between " + str(min) + " and " + str(max) + ": "))

            except:

                print("Not a valid input was given")


        return option
    

    #filters the whole vote data array for objects that match the given expression
    #Format of expression MUST match: x.CountryName == 'Scotland'
    @staticmethod
    def FilterVoteData(data, expression):

        try:
            filter = [x for x in data if eval(expression)]

            return filter
        
        except:
            print("Expression invalid, returning whole data set")
            return data
        

    #filters the whole vote data array for items that contain the given element
    #Format of expression MUST match: "name", "x.Member.firstName"
    @staticmethod
    def SearchVoteData(data, phrase, obj):

        try:
            filter = [x for x in data if phrase in eval(obj)]

            return filter
        
        except:
            print("Expression invalid, returning whole data set")
            return data
        

    # displays all of the data in an easy to read table
    @staticmethod
    def DisplayDataDetails(data):

        tableData = []

        for self in data:

            tableData.append( [ self.ConstituencyName,
                            self.RegionName,
                            self.CountryName,
                            self.ConstituencyType,
                            self.Member.firstName,
                            self.Member.lastName,
                            self.Member.gender,
                            self.Result,
                            self.FirstParty,
                            self.SecondParty,
                            self.Electorate,
                            self.ValidVotes,
                            self.InvalidVotes,
                            self.Majority
                            ] )

        print("\n")
        print(tabulate(tableData, headers=[ 'Constituency Name',
                                            'Region name',
                                            'Country Name',
                                            'Constituency Type', 
                                            'First Name', 
                                            'Last Name', 
                                            'Gender', 
                                            'Result',
                                            'First Party', 
                                            'Second Party', 
                                            'Electorate', 
                                            'Valid Votes', 
                                            'Invalid Votes', 
                                            'Majority'
                                            
                                            ], tablefmt='orgtbl'))