From 5f7a3ddf69719da0146e14e5d6873b6c60421062 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Matt=C3=A9o=20Delabre?= <contact@matteodelabre.me>
Date: Mon, 25 Nov 2019 23:47:27 -0500
Subject: [PATCH] =?UTF-8?q?pageviews:=20Permet=20l=E2=80=99affichage=20?=
 =?UTF-8?q?=C3=A9cran=20ou=20enregistrer?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 data/pageviews/pageviews.py | 30 +++++++++++++++++-------------
 1 file changed, 17 insertions(+), 13 deletions(-)

diff --git a/data/pageviews/pageviews.py b/data/pageviews/pageviews.py
index 17d24cf..1c8f03a 100644
--- a/data/pageviews/pageviews.py
+++ b/data/pageviews/pageviews.py
@@ -10,14 +10,6 @@ from scipy import stats
 import sys
 import mwclient
 
-# Configuration de Matplotlib
-matplotlib.use('pgf')
-matplotlib.rcParams.update({
-    'pgf.texsystem': 'xelatex',
-    'font.family': 'serif',
-    'text.usetex': True,
-    'pgf.rcfonts': False,
-})
 
 # Chemin racine pour les API Wikimedia
 wikimedia_base_path = 'https://wikimedia.org/api/rest_v1'
@@ -223,9 +215,6 @@ def create_year_plot():
         for month in range(1, 13)
     ])
 
-    import locale
-    locale.setlocale(locale.LC_ALL, 'fr_FR.utf8')
-
     ax.set_xticklabels(calendar.month_abbr[1:13])
     return fig, ax
 
@@ -240,7 +229,8 @@ Obtient les statistiques moyenne de vue de pages wiki.
 Paramètres :
 
     project     Projet Wikipédia ciblé (par exemple fr.wikipedia.org).
-    output      Nom du fichier de sortie où sera écrit le graphe.
+    output      Nom du fichier de sortie où sera sauvé le graphe, ou '-' pour
+                afficher le résultat à l’écran.
     article     Nom(s) d’article(s) Wikipédia ciblé(s).
 
 Au moins un article doit être donné. Le nombre de visites est lissé avec un
@@ -255,6 +245,17 @@ dénombrée comme une visite sur la page canonique.
     articles = sys.argv[3:]
 
     site = mwclient.Site(project)
+    output_to_file = output != '-'
+
+    if output_to_file:
+        matplotlib.use('pgf')
+        matplotlib.rcParams.update({
+            'pgf.texsystem': 'xelatex',
+            'font.family': 'serif',
+            'text.usetex': True,
+            'pgf.rcfonts': False,
+        })
+
     fig, ax = create_year_plot()
 
     for article in articles:
@@ -275,4 +276,7 @@ dénombrée comme une visite sur la page canonique.
     fig.autofmt_xdate()
     fig.tight_layout()
 
-    pyplot.savefig(output)
+    if output_to_file:
+        pyplot.savefig(output)
+    else:
+        pyplot.show()