diff options
Diffstat (limited to 'numberparser.py')
| -rw-r--r-- | numberparser.py | 182 |
1 files changed, 182 insertions, 0 deletions
diff --git a/numberparser.py b/numberparser.py new file mode 100644 index 0000000..fb04027 --- /dev/null +++ b/numberparser.py @@ -0,0 +1,182 @@ +#!/usr/bin/env python +# numberparser.py - Translate words to decimal + +# This is part of Kaylee +# -- this code is licensed GPLv3 +# Copyright 2015 Clayton G. Hobbs +# Portions Copyright 2013 Jezra +import re + +# Define the mappings from words to numbers +class NumberParser: + zero = { + 'zero': 0 + } + + ones = { + 'one': 1, + 'two': 2, + 'three': 3, + 'four': 4, + 'five': 5, + 'six': 6, + 'seven': 7, + 'eight': 8, + 'nine': 9 + } + + special_ones = { + 'ten': 10, + 'eleven': 11, + 'twelve': 12, + 'thirteen': 13, + 'fourteen': 14, + 'fifteen': 15, + 'sixteen': 16, + 'seventeen': 17, + 'eighteen': 18, + 'ninteen': 19 + } + + tens = { + 'twenty': 20, + 'thirty': 30, + 'fourty': 40, + 'fifty': 50, + 'sixty': 60, + 'seventy': 70, + 'eighty': 80, + 'ninty': 90 + } + + hundred = { + 'hundred': 100 + } + + exp = { + 'thousand': 1000, + 'million': 1000000, + 'billion': 1000000000 + } + + allowed = [ + 'and' + ] + + def __init__(self): + self.number_words = [] + for word in self.zero: + self.number_words.append(word) + for word in self.ones: + self.number_words.append(word) + for word in self.special_ones: + self.number_words.append(word) + for word in self.tens: + self.number_words.append(word) + for word in self.hundred: + self.number_words.append(word) + for word in self.exp: + self.number_words.append(word) + self.mandatory_number_words = self.number_words.copy() + for word in self.allowed: + self.number_words.append(word) + + def parse_number(self, text_line): + """ + Parse numbers from natural language into ints + + TODO: Throw more exceptions when invalid numbers are detected. Only + allow certian valueless words within numbers. Support zero. + """ + value = 0 + partial_value = 0 + last_list = None + + # Split text_line by commas, whitespace, and hyphens + text_line = text_line.strip() + text_words = re.split(r'[,\s-]+', text_line) + # Parse the number + for word in text_words: + if word in self.zero: + if last_list is not None: + raise ValueError('Invalid number') + value = 0 + last_list = self.zero + elif word in self.ones: + if last_list in (self.zero, self.ones, self.special_ones): + raise ValueError('Invalid number') + value += self.ones[word] + last_list = self.ones + elif word in self.special_ones: + if last_list in (self.zero, self.ones, self.special_ones, self.tens): + raise ValueError('Invalid number') + value += self.special_ones[word] + last_list = self.special_ones + elif word in self.tens: + if last_list in (self.zero, self.ones, self.special_ones, self.tens): + raise ValueError('Invalid number') + value += self.tens[word] + last_list = self.tens + elif word in self.hundred: + if last_list not in (self.ones, self.special_ones, self.tens): + raise ValueError('Invalid number') + value *= self.hundred[word] + last_list = self.hundred + elif word in self.exp: + if last_list in (self.zero, self.exp): + raise ValueError('Invalid number') + partial_value += value * self.exp[word] + value = 0 + last_list = self.exp + elif word not in self.allowed: + raise ValueError('Invalid number') + # Debugging information + #print(word, value, partial_value) + # Finish parsing the number + value += partial_value + return value + + def parse_all_numbers(self, text_line): + nums = [] + t_numless = '' + current_num = '' + + # Split text_line by commas, whitespace, and hyphens + text_line = text_line.strip() + text_words = re.split(r'[,\s-]+', text_line) + for word in text_words: + # If we aren't starting a number, add the word to the result string + if word not in self.mandatory_number_words: + if current_num: + if word in self.number_words: + current_num += word + ' ' + else: + try: + nums.append(self.parse_number(current_num)) + except ValueError: + nums.append(-1) + current_num = '' + t_numless += '%d' + ' ' + if not current_num: + t_numless += word + ' ' + else: + # We're parsing a number now + current_num += word + ' ' + if current_num: + try: + nums.append(self.parse_number(current_num)) + except ValueError: + nums.append(-1) + current_num = '' + t_numless += '%d' + ' ' + + return (t_numless.strip(), nums) + +if __name__ == '__main__': + np = NumberParser() + # Get the words to translate + text_line = input('Enter a number: ') + # Parse it to an integer + value = np.parse_all_numbers(text_line) + # Print the result + print('I claim that you meant the decimal number', value) |