diff --git a/sentiment.py b/sentiment.py index 98bba406f1dfa712d3cf83b114d77903668f605c..4ad6fcb0432df9f584fe9eb3142794557e0ac1cc 100644 --- a/sentiment.py +++ b/sentiment.py @@ -1,6 +1,15 @@ -from __future__ import print_function, division +from enum import Enum -from builtins import input + +class MenuOption(Enum): + SHOW_REVIEW = 'Show a review' + SAVE_TOKEN_LIST = 'Save the token list to a file' + SHOW_DOCUMENT_FREQUENCY = 'Show the document frequency for a particular token' + SHOW_TOKEN_STATISTICS = 'Show all statistics for a particular token' + SHOW_SENTENCE_STATISTICS = 'Show the statistics for a sentence' + SAVE_STOP_WORD_LIST = 'Save the list of stop words to a file' + SHOW_ADJUSTED_SENTENCE_STATISTICS = 'Show the statistics for a sentence with stop words ignored' + EXIT = 'Exit the program' def main():