def main():
""" Main function """
if locale.getdefaultlocale()[0] != 'fr_FR':
locale.setlocale(locale.LC_ALL, 'fr_FR.utf8')
else:
locale.setlocale(locale.LC_ALL, '')
parse_args = ParseArgs.get_method("all")
if parse_args.database == 'compte':
from compta.cli.compte import Compte
compte = Compte(parse_args)
compte.launch_cmd()
if parse_args.database == 'banque':
from compta.cli.banque import Banque
banque = Banque(parse_args)
banque.launch_cmd()
if parse_args.database == 'ecriture':
from compta.cli.ecriture import Ecriture
ecriture = Ecriture(parse_args)
ecriture.launch_cmd()
if parse_args.database == 'montant':
from compta.cli.montant import Montant
montant = Montant(parse_args)
montant.launch_cmd()
if parse_args.database == 'categorie':
from compta.cli.categorie import Categorie
categorie = Categorie(parse_args)
categorie.launch_cmd()
if parse_args.database == 'tag':
from compta.cli.tag import Tag
tag = Tag(parse_args)
tag.launch_cmd()
def set_tag(self, ecriture_id, create=False):
""" Insert a tag if attribute exist and tag name exists """
if self.tag:
if not self.tag_id:
tags = Tag()
tags.attribut = {}
tags.filter = {'nom': self.tag}
tags.get()
if tags.rqst != False:
self.tag_id = tags.rqst[0]['id']
elif create:
tags.attribut['nom'] = self.tag
tags.attribut['valeur'] = ""
tags.create()
self.tag_id = re.match(r"^\/tag\/(?P<tag_id>.*)\/$",
tags.rqst.headers['Location']
).group('tag_id')
self.rqst = RequestServer.post_method("tag_ecriture",
{'ecriture_id': ecriture_id,
'tag_id': self.tag_id
},
)
