Added SFTP Uploading + Now using .env

This commit is contained in:
2025-08-13 21:17:11 +01:00
parent f19c79ac89
commit ee52623b82

18
main.py
View File

@@ -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")