Added SFTP Uploading + Now using .env
This commit is contained in:
18
main.py
18
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")
|
||||
|
||||
Reference in New Issue
Block a user