contentmonster/classes/vessel.py
2021-11-22 11:14:38 +01:00

48 lines
No EOL
1.5 KiB
Python

from classes.connection import Connection
from classes.database import Database
from classes.file import File
from paramiko.ssh_exception import SSHException
import pathlib
class Vessel:
@classmethod
def fromConfig(cls, config):
if "TempDir" in config.keys():
tempdir = config["TempDir"]
else:
tempdir = "/tmp/.ContentMonster/"
if "Address" in config.keys():
return cls(config.name.split()[1], config["Address"], pathlib.Path(tempdir))
else:
raise ValueError("Definition for Vessel " + config.name.split()[1] + " does not contain Address!")
def __init__(self, name: str, address: str, tempdir: pathlib.Path):
self.name = name
self.address = address
self.tempdir = tempdir
self._connection = None
self._uploaded = self.getUploadedFromDB()
@property
def connection(self):
if self._connection:
try:
self._connection._listdir()
except SSHException:
self._connection = None
self._connection = self._connection or Connection(self)
return self._connection
def getUploadedFromDB(self):
db = Database()
return db.getCompletionForVessel(self)
def currentUpload(self):
db = Database()
directory, name, _ = db.getFileByUUID(fileuuid := self.connection.getCurrentUploadUUID())
return File(name, directory, fileuuid)
def clearTempDir(self):
return self.connection.clearTempDir()