130 lines
3.5 KiB
Python
130 lines
3.5 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Script per prelevare dati da MySQL e inviare comandi SITE FTP
|
|
"""
|
|
|
|
import mysql.connector
|
|
from utils.database.connection import connetti_db
|
|
from utils.config import users_loader as setting
|
|
from ftplib import FTP
|
|
import logging
|
|
import sys
|
|
from typing import List, Tuple
|
|
|
|
# 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:
|
|
"""Connessione al server FTP"""
|
|
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:
|
|
logger.error(f"Errore connessione FTP: {e}")
|
|
sys.exit(1)
|
|
|
|
def fetch_data_from_db(connection: mysql.connector.MySQLConnection) -> List[Tuple]:
|
|
"""Preleva i dati dal database"""
|
|
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(f"Prelevate {len(results)} righe dal database")
|
|
return results
|
|
|
|
except mysql.connector.Error as e:
|
|
logger.error(f"Errore query database: {e}")
|
|
return []
|
|
finally:
|
|
cursor.close()
|
|
|
|
def send_site_command(ftp: FTP, command: str) -> bool:
|
|
"""Invia un comando SITE al server FTP"""
|
|
try:
|
|
# Il comando SITE viene inviato usando sendcmd
|
|
response = ftp.sendcmd(f"SITE {command}")
|
|
logger.info(f"Comando SITE '{command}' inviato. Risposta: {response}")
|
|
return True
|
|
except Exception as e:
|
|
logger.error(f"Errore invio comando SITE '{command}': {e}")
|
|
return False
|
|
|
|
def main():
|
|
"""Funzione principale"""
|
|
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(f"Sending ftp command: {ftp_site_command}")
|
|
|
|
# Invia comando SITE
|
|
if send_site_command(ftp_connection, ftp_site_command):
|
|
success_count += 1
|
|
else:
|
|
error_count += 1
|
|
|
|
logger.info(f"Elaborazione completata. Successi: {success_count}, Errori: {error_count}")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Errore generale: {e}")
|
|
|
|
finally:
|
|
# Chiudi connessioni
|
|
try:
|
|
ftp_connection.quit()
|
|
logger.info("Connessione FTP chiusa")
|
|
except Exception as e:
|
|
logger.error(f"Errore chiusura connessione FTP: {e}")
|
|
|
|
try:
|
|
db_connection.close()
|
|
logger.info("Connessione MySQL chiusa")
|
|
except Exception as e:
|
|
logger.error(f"Errore chiusura connessione MySQL: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
main() |