Uploading code again for v1.0

Author: chomes

rsync.py

@@ -1,16 +1,10 @@
 #!/usr/bin/env python3
+# Version 1.0
 # Author: chomes@github
-# Current version: 1.0
-# Version 1.0 basic rsync with copying locally and remotely.
-# Version 1.1 config addition and automation of script.
-# Future versions:
-# Version 1.2 lock file to stop same backup running, along with a way of telling if the backup is still running.
-# Version 1.3 email notification
-# Importing functions
+# importing functions
 import syncfuncs
 from shutil import which
-from pathlib import Path
-import configparser
+
 
 # Checking if user has rsync installed
 if which("rsync") is not None:
@@ -27,30 +21,19 @@
         exit()
 
 
-# If statement to determine manual or automated backup
-config_file = Path("config.ini")
-if config_file.is_file():
-    config = configparser.ConfigParser()
-    config.read("config.ini")
-    if 'Manual' in config:
-        syncfuncs.lsyauto()
-    elif 'LoRem' in config:
-        syncfuncs.loreauto()
-    elif 'RemLo' in config:
-        syncfuncs.reloauto()
-else:
-    print("Config not found, manual backup is starting")
-    # Welcome message
-    print("Welcome to the Backup script, this will allow you to choose whether you want to back up locally or remotely")
-    print("You can choose to either back up from remote to local or local to remote it's entirely your choice")
 
-    # remote or local backup options
-    remlo = input("Is this a back up locally from one place to another or a remote server backup? (local OR remote)")
-    if remlo == 'local':
-        syncfuncs.localsyn()
-    elif remlo == 'remote':
-        resync = input("Are you copying from local to remote server or remote server to local? (local OR remote)")
-        if resync == 'remote':
-            syncfuncs.remlocal()
-        elif resync == 'local':
-            syncfuncs.localrem()
+# Welcome message
+print("Welcome to the Backup script, this will allow you to choose whether you want to back up locally or remotely")
+print("\n")
+print("You can choose to either back up from remote to local or local to remote it's entirely your choice")
+print("\n")
+# remote or local backup options
+remlo = input("Is this a back up local backup or are you backing up to a remote server? (local OR remote) ").lower()
+if remlo == 'local':
+    syncfuncs.localsyn()
+elif remlo == 'remote':
+    resync = input("Is the copy from local or is it from a remote server? (local OR remote) ").lower()
+    if resync == 'remote':
+        syncfuncs.remlocal()
+    elif resync == 'local':
+        syncfuncs.localrem()

syncfuncs.py

@@ -4,31 +4,26 @@
 import time
 import subprocess
 from subprocess import call
-import configparser
 import getpass
 import apt
 
 
-# Function for apt install of package
+# Package for installing apt
 def install_apt(pkg_name):
     cache = apt.Cache()
     cache.update()
     pkg = cache[pkg_name]
     pkg.mark_install()
 
 
-# Function for yum install of package
 def install_yum(pkg_name):
-    rsync_install = ["yum", "install", pkg_name, "-y"]
+    rsync_install = ['yum install %s -y' % (pkg_name)]
     call(rsync_install)
 
 
 # Local sync function
 def sync_call_manual(startbk, sor, des, logloc):
     if startbk == 1:
-        timestamp = time.strftime("%Y%m%d-%H%M")
-        logloc = logloc + timestamp
-        logloc = logloc + "-logfile"
         print("Starting backup, rsync -avv {} {} --log-file={}".format(sor, des, logloc))
         lclresyn = ["rsync", "-avv", sor, des, "--log-file", logloc]
         call(lclresyn)
@@ -116,35 +111,20 @@ def localsyn():
         logging = input("Do you want a log of the backup? (y or n)").lower()
         if logging == "n":
             logloc = ""
+            print("No logs required running backup!")
             startbk = 2
-            print("Saving to config for automation later")
-            config = configparser.ConfigParser()
-            config['Manual'] = {'bkoption': startbk,
-                                'source': sor,
-                                'destination': des,
-                                'log_location': logloc}
-            with open('config.ini', 'w') as configfile:
-                config.write(configfile)
-                configfile.close()
-            print("Config file saved. No logs required running backup!")
             sync_call_manual(startbk, sor, des, logloc)
             break
         elif logging == "y":
             logloc = input("Please type the destination of the log: ")
             if logloc.endswith("/"):
                 conf_log = input("Are you happy with the destination {} ? y or n ".format(logloc)).lower()
                 if conf_log == "y":
+                    timestamp = time.strftime("%Y%m%d-%H%M")
+                    logloc = logloc + timestamp
+                    logloc = logloc + "-logfile"
                     startbk = 1
-                    print("Saving to config for automation later")
-                    config = configparser.ConfigParser()
-                    config['Manual'] = {'bkoption': startbk,
-                                        'source': sor,
-                                        'destination': des,
-                                        'log_location': logloc}
-                    with open('config.ini', 'w') as configfile:
-                        config.write(configfile)
-                        configfile.close()
-                    print("Config saved. Ok, lets run the backup!")
+                    print("Ok lets run the backup!")
                     sync_call_manual(startbk, sor, des, logloc)
                     break
                 else:
@@ -158,10 +138,20 @@ def localsyn():
 # local to remote server backup
 def localrem():
     print("Now we will choose the source and destination")
-    sor = input("What is the FULL PATH to the directory you want to backup? ")
-    des = input("What is the FULL PATH to the directory you want to copy to? ")
-    usn = input('''Please state the user for the backup,
-    if you don't state one we'll use the default one {} '''.format(getpass.getuser()))
+    # Loop to make sure directory ends with /
+    while True:
+        sor = input("What is the FULL PATH to the directory you want to backup? ")
+        if sor.endswith("/"):
+            break
+        else:
+            print("Directory must have a / at the end, please add one")
+    while True:
+        des = input("What is the FULL PATH to the directory you want to copy to? ")
+        if des.endswith("/"):
+            break
+        else:
+            print("Directory must have a / at the end, please add one")
+    usn = input("What's the username? If none given we'll use the current {}".format(getpass.getuser()))
     if not usn:
         usn = getpass.getuser()
     remserv = input("What is the server you want to connect to? ")
@@ -174,21 +164,26 @@ def localrem():
         change = input("Did you want to make any changes? (source, destination, port, username, server, no)").lower()
         if change == 'source':
             sor = input("What is the FULL PATH to the directory you want to backup? ")
-            print("We will copy from {} to {}".format(sor, des))
-            continue
+            if sor.endswith("/"):
+                print("We will copy from {} to {}".format(sor, des))
+                continue
+            else:
+                print("The directory must have a / please add one")
         elif change == 'destination':
             des = input("What is the FULL PATH to the dorectory you want to copy to? ")
-            print("We will copy from {} to {}".format(sor, des))
-            continue
+            if sor.endswith("/"):
+                print("We will copy from {} to {}".format(sor, des))
+                continue
+            else:
+                print("The directory must have a / please add one")
         elif change == 'port':
             servport = input("What is the server port? If you don't choose one we will default to normal ssh ")
             if not servport:
                 servport = 22
             print("The port you have chosen is {}".format(servport))
             continue
         elif change == 'username':
-            usn = input('''Please state the user for the backup, 
-            if you don't state one we'll use the default one {} '''.format(getpass.getuser()))
+            usn = input("What's the username? If none given we'll use the current {}".format(getpass.getuser()))
             print("The username you have chosen is {}".format(usn))
             continue
         elif change == 'server':
@@ -203,41 +198,20 @@ def localrem():
         logging = input("Do you want a log of the backup? (y or n)").lower()
         if logging == "n":
             logloc = ""
+            print("No logs required running backup!")
             startbk = 2
-            print("Saving to config for automation later")
-            config = configparser.ConfigParser()
-            config['LoRem'] = {'bkoption': startbk,
-                               'source': sor,
-                               'destination': des,
-                               'log_location': logloc,
-                               'username': usn,
-                               'remote_server': remserv,
-                               'server_port': servport}
-            with open('config.ini', 'w') as configfile:
-                config.write(configfile)
-                configfile.close()
-            print("Config saved.  No logs required running backup!")
             sync_call_lorem(startbk, sor, des, usn, remserv, servport, logloc)
             break
         elif logging == "y":
             logloc = input("Please type the destination of the log: ")
             if logloc.endswith("/"):
                 conf_log = input("Are you happy with the destination {} ? y or n ".format(logloc)).lower()
                 if conf_log == "y":
+                    timestamp = time.strftime("%Y%m%d-%H%M")
+                    logloc = logloc + timestamp
+                    logloc = logloc + "-logfile"
                     startbk = 1
-                    print("Saving to config for automation later")
-                    config = configparser.ConfigParser()
-                    config['LoRem'] = {'bkoption': startbk,
-                                       'source': sor,
-                                       'destination': des,
-                                       'log_location': logloc,
-                                       'username': usn,
-                                       'remote_server': remserv,
-                                       'server_port': servport}
-                    with open('config.ini', 'w') as configfile:
-                        config.write(configfile)
-                        configfile.close()
-                    print("Config saved.  Ok lets run the backup!")
+                    print("Ok lets run the backup!")
                     sync_call_lorem(startbk, sor, des, usn, remserv, servport, logloc)
                     break
                 else:
@@ -253,8 +227,7 @@ def remlocal():
     print("Now we will choose the source and destination")
     sor = input("What is the FULL PATH to the directory you want to copy from? ")
     des = input("What is the FULL PATH to the directory you want to copy to? ")
-    usn = input('''Please state the user for the backup,
-    if you don't state one we'll use the default one {} '''.format(getpass.getuser()))
+    usn = input("What's the username? If none given we'll use the current {}".format(getpass.getuser()))
     if not usn:
         usn = getpass.getuser()
     remserv = input("What is the server you want to connect to? ")
@@ -280,8 +253,7 @@ def remlocal():
             print("The port you have chosen is {}".format(servport))
             continue
         elif change == 'username':
-            usn = input('''Please state the user for the backup,
-            if you don't state one we'll use the default one {} '''.format(getpass.getuser()))
+            usn = input("What's the username? If none given we'll use the current {}".format(getpass.getuser()))
             print("The username you have chosen is {}".format(usn))
             continue
         elif change == 'server':
@@ -296,100 +268,25 @@ def remlocal():
         logging = input("Do you want a log of the backup? (y or n)").lower()
         if logging == "n":
             logloc = ""
+            print("No logs required running backup!")
             startbk = 3
-            print("Saving to config for automation later")
-            config = configparser.ConfigParser()
-            config['RemLo'] = {'bkoption': startbk,
-                               'source': sor,
-                               'destination': des,
-                               'log_location': logloc,
-                               'username': usn,
-                               'remote_server': remserv,
-                               'server_port': servport}
-            with open('config.ini', 'w') as configfile:
-                config.write(configfile)
-                configfile.close()
-            print("Config saved.  Ok lets run the backup!")
             sync_call_lorem(startbk, sor, des, usn, remserv, servport, logloc)
             break
         elif logging == "y":
             logloc = input("Please type the destination of the log: ")
             if logloc.endswith("/"):
                 conf_log = input("Are you happy with the destination {} ? y or n ".format(logloc)).lower()
                 if conf_log == "y":
+                    timestamp = time.strftime("%Y%m%d-%H%M")
+                    logloc = logloc + timestamp
+                    logloc = logloc + "-logfile"
                     startbk = 4
-                    print("Saving to config for automation later")
-                    config = configparser.ConfigParser()
-                    config['RemLo'] = {'bkoption': startbk,
-                                       'source': sor,
-                                       'destination': des,
-                                       'log_location': logloc,
-                                       'username': usn,
-                                       'remote_server': remserv,
-                                       'server_port': servport}
-                    with open('config.ini', 'w') as configfile:
-                        config.write(configfile)
-                        configfile.close()
-                    print("Config saved.  Ok lets run the backup!")
+                    print("Ok lets run the backup!")
                     sync_call_lorem(startbk, sor, des, usn, remserv, servport, logloc)
                     break
                 else:
                     print("Ok, lets change the destination")
                     continue
             else:
                 print("Destination must end with a / please try again")
-                continue
-
-
-# Automated backup for local
-def lsyauto():
-    print("Reading config")
-    config = configparser.ConfigParser()
-    config.read("config.ini")
-    if config.get("Manual", "bkoption") == '1':
-        startbk = 1
-    else:
-        startbk = 2
-    sor = config.get("Manual", "source")
-    des = config.get("Manual", "destination")
-    logloc = config.get("Manual", "log_location")
-    print("Running backup from {} to {}".format(sor, des))
-    sync_call_manual(startbk, sor, des, logloc)
-
-
-# Automated backup for local to remote
-def loreauto():
-    print("Reading config")
-    config = configparser.ConfigParser()
-    config.read("config.ini")
-    if config.get("LoRem", "bkoption") == '1':
-        startbk = 1
-    else:
-        startbk = 2
-    sor = config.get("LoRem", "source")
-    des = config.get("LoRem", "destination")
-    logloc = config.get("LoRem", "log_location")
-    usn = config.get("LoRem", "username")
-    remserv = config.get("LoRem", "remote_server")
-    servport = config.get("LoRem", "server_port")
-    print("Running backup")
-    sync_call_lorem(startbk, sor, des, usn, remserv, servport, logloc)
-
-
-# Automated backup for remote to local
-def reloauto():
-    print("Reading config")
-    config = configparser.ConfigParser()
-    config.read("config.ini")
-    if config.get("RemLo", "bkoption") == '3':
-        startbk = 3
-    else:
-        startbk = 4
-    sor = config.get("RemLo", "source")
-    des = config.get("RemLo", "destination")
-    logloc = config.get("RemLo", "log_location")
-    usn = config.get("RemLo", "username")
-    remserv = config.get("RemLo", "remote_server")
-    servport = config.get("RemLo", "server_port")
-    print("Running backup")
-    sync_call_lorem(startbk, sor, des, usn, remserv, servport, logloc)
+                continue