alex/python
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

mod 2 pasqua

Browse Source
This commit is contained in:
Alessandro Battilani
2020-04-13 19:24:12 +02:00
parent e85b6949f6
commit 55f0731b4f
6 changed files with 28 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
FtpCsvReceiver.py
View File

@@ -4,7 +4,10 @@ import sys
import os
import pika
import logging
from battilo.timefmt import timestamp_fmt as ts
import argparse
from asebat.timefmt import timestamp_fmt as ts
from pprint import pprint
from pyftpdlib.handlers import FTPHandler
from pyftpdlib.servers import FTPServer
@@ -12,7 +15,7 @@ from pyftpdlib.authorizers import UnixAuthorizer
from pyftpdlib.filesystems import UnixFilesystem
class MyHandler(FTPHandler):
class ASEHandler(FTPHandler):
def on_file_received(self, file):
with open(file, "r") as file_csv:
for i, line in enumerate(file_csv.readlines(4096), 1):
@@ -23,29 +26,45 @@ class MyHandler(FTPHandler):
file_csv.close
path, filename_ext = os.path.split(file)
filename, file_extension = os.path.splitext(filename_ext)
new_path = '/home/' + self.username + '/received/' + unit_name.upper() + '/'
new_filename = new_path + filename + '_' + str(ts.timestamp("tms") + file_extension)
try:
os.mkdir(path + '/' + unit_name.upper())
logging.info("PID {:>5} >> path {} created.".format(os.getpid(), unit_name.upper()))
os.makedirs(new_path)
logging.info("PID {:>5} >> path {} created.".format(os.getpid(), new_path))
except FileExistsError:
logging.info("PID {:>5} >> path {} already exists.".format(os.getpid(), unit_name.upper()))
os.rename(file, path + '/' + unit_name.upper() + '/' + filename + '_' + str(ts.timestamp("tms") + file_extension))
logging.info("PID {:>5} >> path {} already exists.".format(os.getpid(), new_path))
try:
os.rename(file, new_filename)
logging.info("PID {:>5} >> {} moved into {}.".format(os.getpid(), filename_ext, new_filename))
except OSError:
logging.error("PID {:>5} >> Error to move {} into {}.".format(os.getpid(), filename_ext, new_filename))
def on_incomplete_file_received(self, file):
# remove partially uploaded files
os.remove(file)
def main():
parser = argparse.ArgumentParser()
parser.add_argument("-l", "--logfilename", help="log file", required=True)
parser.add_argument("-p", "--port", help="passive port range start", type=int, required=True)
parser.add_argument("-x", "--proxy", help="proxy address", required=True)
args = parser.parse_args()
try:
authorizer = UnixAuthorizer(
rejected_users=["root"], require_valid_shell=True)
handler = MyHandler
handler = ASEHandler
handler.authorizer = authorizer
handler.abstracted_fs = UnixFilesystem
handler.masquerade_address = args.proxy
_range = list(range(args.port, args.port + 20))
handler.passive_ports = _range
logging.basicConfig(format='%(asctime)s %(message)s',
filename='/var/log/ftpCsvRecv.log', level=logging.INFO)
filename="/var/log/" + args.logfilename, level=logging.INFO)
server = FTPServer(('', 21), handler)
server = FTPServer(('0.0.0.0', 21), handler)
server.serve_forever()
except KeyboardInterrupt: