Minor tweak

Author: klaudiosinani

moviebox/data.py

@@ -36,6 +36,22 @@ def splitData(moviesDF, verbose):
     return {'titles': titles, 'categories': categories, 'plots': plots}
 
 
+def validateInput(movieID, recommendationsNumber):
+    # Check whether the input ID is valid
+    if not (isinstance(movieID, int) and movieID >= 0 and movieID <= 4999):
+        printRed('Invalid value for movie ID: "' + str(movieID) + '"')
+        printRed('Input is not a valid Natural number between [0, 4999]')
+        sys.exit(1)
+    # Check whether the recommendations number is valid
+    if not (isinstance(recommendationsNumber, int)
+            and recommendationsNumber >= 1 and recommendationsNumber <= 30):
+        printRed('Invalid value for recommendations number: "' +
+                 str(recommendationsNumber) + '"')
+        printRed('Input is not a valid Natural number between [1, 30]')
+        sys.exit(1)
+    return 0
+
+
 def searchMovie(movieID):
     # Search a movie title by ID
     data = importData(verbose=False)

moviebox/output.py

@@ -1,5 +1,4 @@
 import re
-import sys
 
 from colorama import init
 from termcolor import colored
@@ -67,20 +66,6 @@ def prettyPrint(movieID, title, category, plot, color, showPlots):
         print(colored('  Plot:\n' + plot + '\n', color))
 
 
-def validateInput(movieID, recommendationsNumber):
-    if not (isinstance(movieID, int) and movieID >= 0 and movieID <= 4999):
-        printRed('Invalid value for movie ID: "' + str(movieID) + '"')
-        printRed('Input is not a valid Natural number between [0, 4999]')
-        sys.exit(1)
-    if not (isinstance(recommendationsNumber, int)
-            and recommendationsNumber >= 1 and recommendationsNumber <= 30):
-        printRed('Invalid value for recommendations number: "' +
-                 str(recommendationsNumber) + '"')
-        printRed('Input is not a valid Natural number between [1, 30]')
-        sys.exit(1)
-    return 0
-
-
 def displayHelpMessage():
     # Display cli help message
     print(helpMessage)

moviebox/recommender.py

@@ -1,6 +1,6 @@
-from .data import importData, splitData
+from .data import importData, splitData, validateInput
 from .tfidf import trainEngine, getSimilarities
-from .output import printYellow, printGreen, prettyPrint, validateInput
+from .output import printYellow, printGreen, prettyPrint
 
 green = 'green'  # Green colored text
 yellow = 'yellow'  # Yellow-colored text