diff --git a/main.py b/main.py index e82c87d..1afcd1e 100644 --- a/main.py +++ b/main.py @@ -2,10 +2,20 @@ import openDrive import time import os from imdb import Cinemagoer +import paramiko +from dotenv import load_dotenv +load_dotenv() DVD_DEVICE = "/dev/sr0" +uploadPath = os.getenv('UPLOAD_PATH') + +sftpHost = os.getenv('SFTP_HOST') +sftpPort = os.getenv('SFTP_PORT') +sftpUsername = os.getenv('SFTP_USERNAME') +sftpPassword = os.getenv('SFTP_PASSWORD') + def ripDvd(driveName): print("- DVD has been detected") print("- Searching IMDB for " + driveName) @@ -20,6 +30,14 @@ def ripDvd(driveName): print("- Will now procede to ripping DVD") movieName = driveName os.system('HandBrakeCLI -i /dev/sr0 -o output/"' + movieName + '.mp4" --preset-import-file preset.json') + sftpTransport = paramiko.Transport((sftpHost, sftpPort)) + sftpTransport.connect(username = sftpUsername, password = sftpPassword) + sftp = paramiko.SFTPClient.from_transport(sftpTransport) + print("- Movie has been ripped, will now procede to upload file") + sftp.put('output/"' + movieName + '.mp4"', uploadPath + movieName + '.mp4') + sftp.close() + sftpTransport.close() + print("- Movie has been uploaded") def waitForDvd(): print("- Insert a DVD into drive")