285 lines
7.7 KiB
Python
285 lines
7.7 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import csv
|
|
import argparse
|
|
from clint.textui import puts, colored, indent
|
|
from clint.textui import columns
|
|
#import pytz
|
|
#import tzlocal
|
|
import time
|
|
from datetime import datetime, timedelta, date
|
|
import dateutil.parser
|
|
from dateutil.rrule import rrulestr
|
|
#from icalendar import Calendar,Event,Todo,Journal,Alarm
|
|
#import caldav
|
|
#import uuid
|
|
#import json
|
|
import os
|
|
#os.environ['LANGUAGE']='fr'
|
|
#import logging
|
|
import sys
|
|
#import re
|
|
#import urllib3
|
|
#from getpass import getpass
|
|
#from sympy import symbols
|
|
#from sympy.plotting import textplot
|
|
import subprocess
|
|
from subprocess import Popen, PIPE
|
|
import numpy as np
|
|
import gettext
|
|
#gettext.bindtextdomain('base', 'locale/en/LC_MESSAGES/base.mo')
|
|
#gettext.textdomain('base')
|
|
# Voir : https://docs.python.org/fr/3/library/gettext.html
|
|
en = gettext.translation('base', localedir='locale/', languages=['en'])
|
|
en.install()
|
|
_ = en.gettext # english
|
|
# fr = gettext.translation('base', localedir='locale/', languages=['fr'])
|
|
# fr.install()
|
|
# _ = fr.gettext # english
|
|
# Voir : https://medium.com/i18n-and-l10n-resources-for-developers/how-to-translate-python-applications-with-the-gnu-gettext-module-5c1c085041
|
|
|
|
"""
|
|
Une mesure est caractérisée par :
|
|
- sa valeur
|
|
- son type
|
|
- sa date
|
|
- sa métrique
|
|
- son identifiant
|
|
- un commentaire
|
|
- autre
|
|
"""
|
|
|
|
def save_file_data():
|
|
""" Enregistrement du nom du fichier de données """
|
|
f = open("Fonctionnel.txt", "r")
|
|
contents = f.readlines()
|
|
f.close()
|
|
linenum = 0
|
|
for row in contents:
|
|
if row.split(' ',1)[0] == 'plot':
|
|
contents.insert(linenum, "plot '"+args.data+"' using 3:1 pt 'X' title ''\n")
|
|
del(contents[linenum+1])
|
|
linenum += 1
|
|
f = open("Fonctionnel.txt", "w")
|
|
contents = "".join(contents)
|
|
f.write(contents)
|
|
f.close()
|
|
|
|
|
|
def save_mesure():
|
|
""" Affichage et enregistrement d'une mesure """
|
|
print("Votre mesure a été enregistrée : ")
|
|
with indent(4, quote=colored.blue('')):
|
|
puts(colored.green("-votre poids : {weight} kg").format(weight=args.weight))
|
|
puts("-son type : {typ}".format(typ=args.type))
|
|
puts("-la date : {dat}".format(dat=args.date))
|
|
puts("-la métrique : {metric}".format(metric=args.metric))
|
|
puts("-l'identifiant : {ide}".format(ide=args.id))
|
|
puts("-votre commentaire : {comment}".format(comment=args.comment))
|
|
puts("-autre : {other}".format(other=args.other))
|
|
with open(str(args.data), 'a', newline='\n') as csvfile:
|
|
weightwriter = csv.writer(csvfile, delimiter='|', quotechar='"', quoting=csv.QUOTE_MINIMAL)
|
|
weightwriter.writerow([str(args.weight), \
|
|
str(args.type), \
|
|
str(datetime.now().strftime("%Y-%m-%d %H:%M:%S")), \
|
|
str(args.metric), str(args.id), \
|
|
str(args.comment), \
|
|
str(args.other)])
|
|
|
|
def read_measures():
|
|
""" Affichage de l'ensemble des mesures """
|
|
data_file = open(str(args.data), 'r')
|
|
data = csv.reader(data_file, delimiter='|')
|
|
linenum = 0
|
|
for row in data:
|
|
with indent(2, quote=colored.blue('')):
|
|
puts(columns([(colored.red(str(linenum))), 4], \
|
|
[(colored.green(str(row[0])+' kg')), 15], \
|
|
[(str(row[1])),10], \
|
|
[(colored.yellow(str(row[2]))),25], \
|
|
[(str(row[3])),6], \
|
|
[(str(row[4])),4], \
|
|
[(colored.magenta(str(row[5]))),20])), \
|
|
[(str(row[6])),10]
|
|
linenum += 1
|
|
data_file.close()
|
|
|
|
def remove_measure():
|
|
""" Remove a measure """
|
|
f = open(args.data, "r")
|
|
contents = f.readlines()
|
|
f.close()
|
|
# remove the line item from list, by line number, starts from 0
|
|
store_temp = str(args.remove)+' '+contents[args.remove]
|
|
g = open('remove.txt', "w")
|
|
g.write(str(store_temp))
|
|
g.close()
|
|
contents.pop(args.remove)
|
|
|
|
f = open(args.data, "w")
|
|
contents = "".join(contents)
|
|
f.write(contents)
|
|
f.close()
|
|
|
|
def undo_remove():
|
|
""" Undo remove the last line removed """
|
|
f = open(args.data, "r")
|
|
contents = f.readlines()
|
|
f.close()
|
|
g = open('remove.txt', 'r')
|
|
store_temp = g.readline().split(' ',1)
|
|
g.close()
|
|
# insert the line item from list, by its line number
|
|
contents.insert(int(store_temp[0]), store_temp[1])
|
|
|
|
f = open(args.data, "w")
|
|
contents = "".join(contents)
|
|
f.write(contents)
|
|
f.close()
|
|
|
|
def trace_graph():
|
|
""" Trace le graphe de l'évolution du poids avec Gnuplot """
|
|
if args.dates != None:
|
|
print(args.dates)
|
|
# set xrange [*:*]
|
|
f = open("Fonctionnel.txt", "r")
|
|
contents = f.readlines()
|
|
f.close()
|
|
linenum = 0
|
|
for row in contents:
|
|
motif = row.split(' ',2)[0]+' '+row.split(' ',2)[1]
|
|
if motif == 'set xrange':
|
|
if args.dates[0] == '*':
|
|
ligne = "set xrange [*:'"+args.dates[1]+"']\n"
|
|
elif args.dates[1] == '*':
|
|
ligne = "set xrange ['"+args.dates[0]+"':*]\n"
|
|
else:
|
|
ligne = "set xrange ['"+args.dates[0]+"':'"+args.dates[1]+"']\n"
|
|
contents.insert(linenum, ligne)
|
|
del(contents[linenum+1])
|
|
linenum += 1
|
|
f = open("Fonctionnel.txt", "w")
|
|
contents = "".join(contents)
|
|
f.write(contents)
|
|
f.close()
|
|
else:
|
|
f = open("Fonctionnel.txt", "r")
|
|
contents = f.readlines()
|
|
f.close()
|
|
linenum = 0
|
|
for row in contents:
|
|
motif = row.split(' ',2)[0]+' '+row.split(' ',2)[1]
|
|
if motif == 'set xrange':
|
|
contents.insert(linenum, "set xrange [*:*]\n")
|
|
del(contents[linenum+1])
|
|
linenum += 1
|
|
f = open("Fonctionnel.txt", "w")
|
|
contents = "".join(contents)
|
|
f.write(contents)
|
|
f.close()
|
|
#process = Popen(['gnuplot', 'Fonctionnel.txt'], stdout=PIPE, stderr=PIPE)
|
|
process = Popen(['gnuplot', 'Fonctionnel.txt'])#, stdout=PIPE, stderr=PIPE)
|
|
stdout, stderr = process.communicate()
|
|
#print(stdout)
|
|
|
|
|
|
def tests_subproc():
|
|
""" Test du module subprocess """
|
|
process = Popen(['cat', 'data2.csv'], stdout=PIPE, stderr=PIPE)
|
|
stdout, stderr = process.communicate()
|
|
print(stdout)
|
|
|
|
### Main ###
|
|
""" Programme de relevé de poids
|
|
Attention, le fichier data.csv à une ligne d'entête à laquelle
|
|
il manque un séparateur | final pour que le nombre de champs soit
|
|
correct.
|
|
"""
|
|
|
|
# Création du parseur pour les arguments
|
|
parser = argparse.ArgumentParser()
|
|
texte = _('A script to manage your weight in command line.')
|
|
parser.description=texte
|
|
# Création des arguments
|
|
parser.add_argument('-da', '--data',
|
|
default='data2.csv',
|
|
help=_('optional : file data')
|
|
)
|
|
parser.add_argument('-w', '--weight',
|
|
type=float,
|
|
default=-1,
|
|
help=_('optional : weight')
|
|
)
|
|
parser.add_argument('-t', '--type',
|
|
default='WEIGHT',
|
|
help=_('optional : type')
|
|
)
|
|
parser.add_argument('-d', '--date',
|
|
default=datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
|
help=_('optional : date')
|
|
)
|
|
parser.add_argument('-i', '--id',
|
|
default='',
|
|
help=_('optional : id')
|
|
)
|
|
parser.add_argument('-m', '--metric',
|
|
default='true',
|
|
help=_('optional : metric')
|
|
)
|
|
parser.add_argument('-c', '--comment',
|
|
default=' ',
|
|
help=_('optional : comment')
|
|
)
|
|
parser.add_argument('-o', '--other',
|
|
default='',
|
|
help=_('optional : other')
|
|
)
|
|
parser.add_argument('-r', '--read',
|
|
help='optional read measures',
|
|
action='store_true'
|
|
)
|
|
parser.add_argument('-rem', '--remove',
|
|
type=int,
|
|
default='-1',
|
|
help=_('optional : delete a measure by its number in the list')
|
|
)
|
|
parser.add_argument('-g', '--graph',
|
|
help=_('optional : plot graph'),
|
|
action='store_true'
|
|
)
|
|
parser.add_argument('-ds', '--dates',
|
|
nargs='+',
|
|
help=_("optional : min_date max_date -> 'Y-m-d hh:mm:ss' or '*' for an automatic date.")
|
|
)
|
|
parser.add_argument('-tst', '--tests',
|
|
help='optional tests operations',
|
|
action='store_true'
|
|
)
|
|
parser.add_argument('-ur', '--undoremove',
|
|
help=_('optional : undo remove the last line removed'),
|
|
action='store_true'
|
|
)
|
|
parser.add_argument('-v', '--version',
|
|
version='0.0.5',
|
|
action='version'
|
|
)
|
|
# Récupération des arguments
|
|
args = parser.parse_args()
|
|
# Lancement des actions
|
|
if args.data != 'data2.csv':
|
|
save_file_data()
|
|
if args.weight != -1:
|
|
save_mesure()
|
|
if args.read:
|
|
read_measures()
|
|
if args.remove >= 0:
|
|
remove_measure()
|
|
if args.graph:
|
|
trace_graph()
|
|
if args.tests:
|
|
tests_subproc()
|
|
if args.undoremove:
|
|
undo_remove()
|
|
|