Update skyhook

Author: deedeecx330

skyhook

@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 
-import sys, db, updown, config
+import os, sys, db, updown, config
 
 def main(command, target):
 
@@ -25,7 +25,9 @@ def main(command, target):
             print("[+] Successfully exported history to {}".format(target))
 
     elif command == "save":
+        workDir = os.getcwd() 
         for tar in target.split(','):
+            os.chdir(workDir)
             res = db.saveOne(tar)
             if res == 1:
                 print("[!] Could not find {} in local history".format(tar))
@@ -35,7 +37,9 @@ def main(command, target):
                 print("[+] Successfully saved {} to local directory".format(tar))
 
     elif command == "import":
+        workDir = os.getcwd() 
         for tar in target.split(','):
+            os.chdir(workDir)
             res = db.importDb(tar)
             if res == 1:
                 print("[!] Could not read {}".format(tar))
@@ -47,7 +51,9 @@ def main(command, target):
                 print("[+] Successfully imported {}".format(tar))
 
     elif command == "delete":
+        workDir = os.getcwd() 
         for tar in target.split(','):
+            os.chdir(workDir)
             res = db.deleteItem(tar)
             if res == 1:
                 print("[!] Could not find {} in history".format(tar))
@@ -57,7 +63,9 @@ def main(command, target):
                 print("[+] Successfully deleted {} from history".format(tar))
 
     elif command == "upload":
+        workDir = os.getcwd() 
         for tar in target.split(','):
+            os.chdir(workDir)
             res = updown.uploadFile(tar)
             if res == 1:
                 print("[!] Could not find {}".format(tar))
@@ -71,7 +79,9 @@ def main(command, target):
                 print("[+] Successfully uploaded {}".format(tar))
 
     elif command == "download":
+        workDir = os.getcwd() 
         for tar in target.split(','):
+            os.chdir(workDir)
             res = updown.downloadFile(tar)
             if res == 1:
                 print("[!] Could not retrieve {} from history".format(tar))