40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
|
import os
|
||
|
|
import shutil
|
||
|
|
from log_picker.sending.senderbaseclass import SenderBaseClass
|
||
|
|
from log_picker.sending.senderbaseclass import SenderError
|
||
|
|
|
||
|
|
|
||
|
|
class LocalSender(SenderBaseClass):
|
||
|
|
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
SenderBaseClass.__init__(self, args, kwargs)
|
||
|
|
self.path = None
|
||
|
|
|
||
|
|
def set_path(self, directory):
|
||
|
|
self.path = directory
|
||
|
|
|
||
|
|
if os.path.exists(self.path) and not os.path.isdir(self.path):
|
||
|
|
raise SenderError('Cannot create "%s" directory. A file of '
|
||
|
|
'the same name already exists.' % self.path)
|
||
|
|
|
||
|
|
|
||
|
|
def sendfile(self, filename, contenttype):
|
||
|
|
|
||
|
|
# Another check due possibility of race condition
|
||
|
|
if os.path.exists(self.path):
|
||
|
|
if not os.path.isdir(self.path):
|
||
|
|
raise SenderError('Cannot create "%s" directory. A file of '
|
||
|
|
'the same name already exists.' % self.path)
|
||
|
|
else:
|
||
|
|
try:
|
||
|
|
os.makedirs(self.path)
|
||
|
|
except Exception as e:
|
||
|
|
raise SenderError('Cannot create "%s" directory: %s' % \
|
||
|
|
(self.path, e))
|
||
|
|
try:
|
||
|
|
shutil.copy(filename, self.path)
|
||
|
|
except Exception as e:
|
||
|
|
raise SenderError('Could not save "%s" to "%s": %s' % \
|
||
|
|
(os.path.basename(filename), self.path, e))
|
||
|
|
|