Skip to content
Extraits de code Groupes Projets
cli.py 5,97 ko
Newer Older
Bastien Le Querrec's avatar
Bastien Le Querrec a validé
import argparse
import logging
Bastien Le Querrec's avatar
Bastien Le Querrec a validé

from RAAspotter import RAAspotter
Bastien Le Querrec's avatar
Bastien Le Querrec a validé

# Config
__KEYWORDS = os.getenv('KEYWORDS') or \
    'vidéoprotection,caméras,captation,aéronef'
__DATA_DIR_ROOT = os.path.dirname(os.path.abspath(__file__)) + '/data/'
__SMTP_HOSTNAME = os.getenv('SMTP_HOSTNAME') or 'localhost'
__SMTP_USERNAME = os.getenv('SMTP_USERNAME') or ''
__SMTP_PASSWORD = os.getenv('SMTP_PASSWORD') or ''
__EMAIL_FROM = os.getenv('EMAIL_FROM')
__EMAIL_TO = os.getenv('EMAIL_TO')
if os.getenv('SMTP_PORT'):
    __SMTP_PORT = int(os.getenv('SMTP_PORT'))
    __SMTP_PORT = 587
if os.getenv('SMTP_STARTTLS'):
    __SMTP_STARTTLS = True
    __SMTP_STARTTLS = False
if os.getenv('SMTP_SSL'):
    __SMTP_SSL = True
    __SMTP_SSL = False
    __NOT_BEFORE = datetime.datetime.strptime(
        os.getenv('NOT_BEFORE'), '%Y-%m-%d'
    )
    __NOT_BEFORE = datetime.datetime(2024, 1, 1)
__MASTODON_ACCESS_TOKEN = os.getenv('MASTODON_ACCESS_TOKEN')
__MASTODON_INSTANCE = os.getenv('MASTODON_INSTANCE')
Bastien Le Querrec's avatar
Bastien Le Querrec a validé

# Liste des préfectures supportées
available_prefs = [
    'ppparis',
    'pref04',
    'pref06',
    'pref13',
    'pref34',
    'pref35',
    'pref38',
    'pref59',
    'pref62',
    'pref65',
    'pref69',
    'pref83',
    'pref976'
Bastien Le Querrec's avatar
Bastien Le Querrec a validé
# Début du script
parser = argparse.ArgumentParser(
    prog='cli.py',
    description='Télécharge les RAA d\'une préfecture donnée et recherche des mots-clés'
)
parser.add_argument(
    '-p',
    '--pref',
    action='store',
    help='identifiant de la préfecture',
    required=True,
    choices=available_prefs
)
parser.add_argument(
    '-k',
    '--keywords',
    action='store',
    help='liste des termes recherchés, séparés par une virgule (par défaut : vidéoprotection,caméras,captation,aéronef)'
)
parser.add_argument(
    '--not-before',
    action='store',
    help='n\'analyse pas les RAA datant d\'avant la date indiquée, au format YYYY-MM-DD (par défaut : 2024-01-01)'
)
parser.add_argument(
    '--smtp-hostname',
    action='store',
    help='nom d\'hôte SMTP (par défaut : localhost)'
)
parser.add_argument(
    '--smtp-username',
    action='store',
    help='nom d\'utilisateur SMTP (par défaut : vide)'
)
parser.add_argument(
    '--smtp-password',
    action='store',
    help='mot de passe SMTP (par défaut : vide)'
)
parser.add_argument(
    '--smtp-port',
    action='store',
    help='port SMTP (par défaut : 587)'
)
parser.add_argument(
    '--smtp-starttls',
    action='store_true',
    help='connexion SMTP avec STARTTLS'
)
parser.add_argument(
    '--smtp-ssl',
    action='store_true',
    help='connexion SMTP avec SSL'
)
parser.add_argument(
    '-f',
    '--email-from',
    action='store',
    help='adresse de courrier électronique expéditrice des notifications'
)
parser.add_argument(
    '-t',
    '--email-to',
    action='store',
    help='adresses de courrier électronique destinataires des notifications (séparées par une virgule)'
    parser.add_argument(
        f'--{pref}-email-to',
        action='store',
        help=f'adresses de courrier électronique destinataires des notifications (séparées par une virgule) uniquement si l\'analyse concerne {pref} (s\'ajoute à celles précisées dans --email-to)'
    )

parser.add_argument(
    '--mastodon-access-token',
    action='store',
    help='jeton d\'accès pour publier sur Mastodon (par défaut : vide)'
)
parser.add_argument(
    '--mastodon-instance',
    action='store',
    help='URL de l\'instance (doit inclure "http://" ou "https://" ; par défaut : vide)'
)
parser.add_argument(
    '-v',
    action='store_true',
    help='relève le niveau de verbosité à INFO'
)
parser.add_argument(
    '-vv',
    action='store_true',
    help='relève le niveau de verbosité à DEBUG'
)
Bastien Le Querrec's avatar
Bastien Le Querrec a validé
args = parser.parse_args()

    logging.basicConfig(level=logging.INFO)
Bastien Le Querrec's avatar
Bastien Le Querrec a validé

    logging.basicConfig(level=logging.DEBUG)
Bastien Le Querrec's avatar
Bastien Le Querrec a validé

if args.keywords:
    __KEYWORDS = args.keywords
Bastien Le Querrec's avatar
Bastien Le Querrec a validé

    __NOT_BEFORE = datetime.datetime.strptime(args.not_before, '%Y-%m-%d')
if args.smtp_hostname:
    __SMTP_HOSTNAME = args.smtp_hostname

if args.smtp_username:
    __SMTP_USERNAME = args.smtp_username

if args.smtp_password:
    __SMTP_PASSWORD = args.smtp_password

if args.smtp_port:
    __SMTP_PORT = int(args.smtp_port)

if args.smtp_starttls:
    __SMTP_STARTTLS = True

if args.smtp_ssl:
    __SMTP_SSL = True

if args.email_from:
    __EMAIL_FROM = args.email_from

if args.email_to:
    __EMAIL_TO = args.email_to
if args.mastodon_access_token:
    __MASTODON_ACCESS_TOKEN = args.mastodon_access_token

if args.mastodon_instance:
    __MASTODON_INSTANCE = args.mastodon_instance
__DATA_DIR = f'{__DATA_DIR_ROOT}{args.pref}/'

# On calcule la liste des mails à notifier (liste générale EMAIL_TO + liste
# prefecture EMAIL_TO_PREF**)
__PREF_EMAIL_TO = ''
pref_var_name = f'{args.pref}_EMAIL_TO'.upper()
if os.getenv(pref_var_name):
    __PREF_EMAIL_TO = os.getenv(pref_var_name)
    for arg in vars(args).items():
        if arg[0] == f'{args.pref}_email_to':
            __PREF_EMAIL_TO = arg[1]
    __EMAIL_TO = f'{__EMAIL_TO},{__PREF_EMAIL_TO}'
Bastien Le Querrec's avatar
Bastien Le Querrec a validé
# On crée le dossier de téléchargement
os.makedirs(__DATA_DIR, exist_ok=True)

module = importlib.import_module(f'RAAspotter_{args.pref}')
raa_spotter = getattr(module, f'RAAspotter_{args.pref}')(__DATA_DIR)

raa_spotter.not_before = __NOT_BEFORE
raa_spotter.configure_mailer(__SMTP_HOSTNAME, __SMTP_USERNAME, __SMTP_PASSWORD, __SMTP_PORT, __SMTP_STARTTLS, __SMTP_SSL,
                             __EMAIL_FROM, __EMAIL_TO, f'[RAAspotter] [{raa_spotter.full_name}] Nouveaux éléments trouvés')
raa_spotter.configure_mastodon(__MASTODON_ACCESS_TOKEN, __MASTODON_INSTANCE,
                               f'[{raa_spotter.full_name}]', f'#{raa_spotter.short_code}')