From 8b3d083938d438b32d4290b6d9851fec4bcb62e6 Mon Sep 17 00:00:00 2001
From: Kevin Midboe <kevin.midboe@gmail.com>
Date: Thu, 18 Oct 2018 20:27:03 +0200
Subject: [PATCH] Handles input from argv as path.

---
 src/cli.py | 46 +++++++++++++++++++++++++++++-----------------
 1 file changed, 29 insertions(+), 17 deletions(-)

diff --git a/src/cli.py b/src/cli.py
index 0abf09f..e1bc7bb 100755
--- a/src/cli.py
+++ b/src/cli.py
@@ -1,25 +1,37 @@
 #!usr/bin/env python3.6
+from guessit import guessit
+import click
 
 from core import scan_folder, moveHome 
 from video import Video
-from guessit import guessit
-
 from exceptions import InsufficientInfoError
 
-videos, insufficient_info = scan_folder('Spider.Man')
-print('Sweet lemonade: {} {}'.format(videos, insufficient_info))
+@click.command()
+@click.argument('path')
+def main(path):
+    videos, insufficient_info = scan_folder(path)
+    # print('Sweet lemonade: {} {}'.format(videos, insufficient_info))
 
-for video in videos:
-    moveHome(video)
-
-while len(insufficient_info) > 1:
-    for file in insufficient_info:
-        supplementary_info = input("Insufficient info for match file: '{}'\nSupplementary info: ".format(file)) 
-        print(supplementary_info)
-        try:
-            video = Video.fromguess(file, guessit(supplementary_info))
-            insufficient_info.pop()
-        except InsufficientInfoError:
-            pass
-            
+    for video in videos:
         moveHome(video)
+
+    while len(insufficient_info) >= 1:
+        for file in insufficient_info:
+            supplementary_info = input("Insufficient info for match file: '{}'\nSupplementary info: ".format(file)) 
+
+            if supplementary_info is 'q':
+                exit(0)
+            if supplementary_info is 's':
+                insufficient_info.pop()
+                continue
+
+            try:
+                video = Video.fromguess(file, guessit(supplementary_info))
+                print(video)
+                moveHome(video)
+                insufficient_info.pop()
+            except InsufficientInfoError:
+                pass
+            
+if __name__ == '__main__':
+    main()