152 lines
4.4 KiB
Python
152 lines
4.4 KiB
Python
#!.venv/bin/python
|
|
"""
|
|
Script per prelevare dati da MySQL e inviare comandi SITE FTP
|
|
"""
|
|
|
|
from ftplib import FTP
|
|
import logging
|
|
import sys
|
|
from typing import List, Tuple
|
|
import mysql.connector
|
|
from utils.database.connection import connetti_db
|
|
from utils.config import users_loader as setting
|
|
|
|
|
|
# Configurazione logging
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(levelname)s - %(message)s'
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Configurazione server FTP
|
|
FTP_CONFIG = {
|
|
'host': 'localhost',
|
|
'user': 'admin',
|
|
'password': 'batt1l0',
|
|
'port': 2121
|
|
}
|
|
|
|
def connect_ftp() -> FTP:
|
|
"""
|
|
Establishes a connection to the FTP server using the predefined configuration.
|
|
Returns:
|
|
FTP: An active FTP connection object.
|
|
"""
|
|
try:
|
|
ftp = FTP()
|
|
ftp.connect(FTP_CONFIG['host'], FTP_CONFIG['port'])
|
|
ftp.login(FTP_CONFIG['user'], FTP_CONFIG['password'])
|
|
logger.info("Connessione FTP stabilita")
|
|
return ftp
|
|
except Exception as e: # pylint: disable=broad-except
|
|
logger.error("Errore connessione FTP: %s", e)
|
|
sys.exit(1)
|
|
|
|
def fetch_data_from_db(connection: mysql.connector.MySQLConnection) -> List[Tuple]:
|
|
"""
|
|
Fetches username and password data from the 'ftp_accounts' table in the database.
|
|
|
|
Args:
|
|
connection (mysql.connector.MySQLConnection): The database connection object.
|
|
Returns:
|
|
List[Tuple]: A list of tuples, where each tuple contains (username, password).
|
|
"""
|
|
try:
|
|
cursor = connection.cursor()
|
|
|
|
# Modifica questa query secondo le tue esigenze
|
|
query = """
|
|
SELECT username, password
|
|
FROM ase_lar.ftp_accounts
|
|
"""
|
|
|
|
cursor.execute(query)
|
|
results = cursor.fetchall()
|
|
|
|
logger.info("Prelevate %s righe dal database", len(results))
|
|
return results
|
|
|
|
except mysql.connector.Error as e:
|
|
logger.error("Errore query database: %s", e)
|
|
return []
|
|
finally:
|
|
cursor.close()
|
|
|
|
def send_site_command(ftp: FTP, command: str) -> bool:
|
|
"""
|
|
Sends a SITE command to the FTP server.
|
|
|
|
Args:
|
|
ftp (FTP): The FTP connection object.
|
|
command (str): The SITE command string to send (e.g., "ADDU username password").
|
|
Returns:
|
|
bool: True if the command was sent successfully, False otherwise.
|
|
"""
|
|
try:
|
|
# Il comando SITE viene inviato usando sendcmd
|
|
response = ftp.sendcmd(f"SITE {command}")
|
|
logger.info("Comando SITE %s inviato. Risposta: %s", command, response)
|
|
return True
|
|
except Exception as e: # pylint: disable=broad-except
|
|
logger.error("Errore invio comando SITE %s: %s", command, e)
|
|
return False
|
|
|
|
def main():
|
|
"""
|
|
Main function to connect to the database, fetch FTP user data, and send SITE ADDU commands to the FTP server.
|
|
"""
|
|
logger.info("Avvio script caricamento utenti FTP")
|
|
cfg = setting.Config()
|
|
|
|
# Connessioni
|
|
db_connection = connetti_db(cfg)
|
|
ftp_connection = connect_ftp()
|
|
|
|
try:
|
|
# Preleva dati dal database
|
|
data = fetch_data_from_db(db_connection)
|
|
|
|
if not data:
|
|
logger.warning("Nessun dato trovato nel database")
|
|
return
|
|
|
|
success_count = 0
|
|
error_count = 0
|
|
|
|
# Processa ogni riga
|
|
for row in data:
|
|
username, password = row
|
|
|
|
# Costruisci il comando SITE completo
|
|
ftp_site_command = f'addu {username} {password}'
|
|
|
|
logger.info("Sending ftp command: %s", ftp_site_command)
|
|
|
|
# Invia comando SITE
|
|
if send_site_command(ftp_connection, ftp_site_command):
|
|
success_count += 1
|
|
else:
|
|
error_count += 1
|
|
|
|
logger.info("Elaborazione completata. Successi: %s, Errori: %s", success_count, error_count)
|
|
|
|
except Exception as e: # pylint: disable=broad-except
|
|
logger.error("Errore generale: %s", e)
|
|
|
|
finally:
|
|
# Chiudi connessioni
|
|
try:
|
|
ftp_connection.quit()
|
|
logger.info("Connessione FTP chiusa")
|
|
except Exception as e: # pylint: disable=broad-except
|
|
logger.error("Errore chiusura connessione FTP: %s", e)
|
|
|
|
try:
|
|
db_connection.close()
|
|
logger.info("Connessione MySQL chiusa")
|
|
except Exception as e: # pylint: disable=broad-except
|
|
logger.error("Errore chiusura connessione MySQL: %s", e)
|
|
|
|
if __name__ == "__main__":
|
|
main() |