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 time
|
||||||
import os
|
import os
|
||||||
from imdb import Cinemagoer
|
from imdb import Cinemagoer
|
||||||
|
import paramiko
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
|
||||||
|
load_dotenv()
|
||||||
|
|
||||||
DVD_DEVICE = "/dev/sr0"
|
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):
|
def ripDvd(driveName):
|
||||||
print("- DVD has been detected")
|
print("- DVD has been detected")
|
||||||
print("- Searching IMDB for " + driveName)
|
print("- Searching IMDB for " + driveName)
|
||||||
@@ -20,6 +30,14 @@ def ripDvd(driveName):
|
|||||||
print("- Will now procede to ripping DVD")
|
print("- Will now procede to ripping DVD")
|
||||||
movieName = driveName
|
movieName = driveName
|
||||||
os.system('HandBrakeCLI -i /dev/sr0 -o output/"' + movieName + '.mp4" --preset-import-file preset.json')
|
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():
|
def waitForDvd():
|
||||||
print("- Insert a DVD into drive")
|
print("- Insert a DVD into drive")
|
||||||
|
|||||||
Reference in New Issue
Block a user