def listachat(self, data, user_data, history, for_user):
session = DBSession()
today = datetime.date.today()
if history < 1:
send_error(data['channel'],
"Non non non, history doit être > 1 <@{0}> ;)".format(
user_data['id']),
markdown=True)
return
user = None
if for_user:
user = self.get_db_user(session, for_user)
if not user:
send_error(data['channel'], 'Erreur zjaifnazgoizangoiazg')
delta = datetime.timedelta(days=history)
start_date = today - delta
query = session.query(Expense).filter(Expense.date >= start_date)
if user:
query = query.filter(Expense.user_id == user.id)
expenses = query.all()
outputs = []
if not expenses:
if user:
outputs.append(
"Pas d'achat dans les {0} derniers jours pour {1}.".format(
history, user.name))
else:
outputs.append(
"Pas d'achat dans les {0} derniers jours.".format(history))
else:
if user:
outputs.append(
"Liste des achats depuis {0} jours par {1} :".format(
history, user.name))
else:
outputs.append(
"Liste des achats depuis {0} jours :".format(history))
for expense in expenses:
if user:
text = "- Le {0} : {1} €".format(expense.date,
expense.amount)
else:
text = "- {0}, le {1} : {2} €".format(
expense.user.name, expense.date, expense.amount)
if expense.description:
text += " pour {0}".format(expense.description)
outputs.append(text)
send_info(data['channel'],
text='\n'.join(outputs),
markdown=True,
thread=data["ts"])
def mealprice(self, data, user_data):
session = DBSession()
total_expenses = 0.0
expenses = session.query(Expense).all()
for expense in expenses:
total_expenses += expense.amount
total_presences = 0
presences = session.query(Presence).all()
for presence in presences:
total_presences += presence.meals
average_price = total_expenses / total_presences
outputs = []
outputs.append("Total des dépenses : {0} €".format(total_expenses))
outputs.append("Total de repas : {0}".format(total_presences))
outputs.append("Prix moyen d'un repas : {0} €".format(average_price))
send_info(data['channel'], text='\n'.join(outputs), markdown=True)
def sync():
"""Sync user with slack."""
endpoint = config.get('default', 'endpoint')
bot_token = config.get(endpoint, 'bot_token')
slack_client = SlackClient(bot_token)
session = DBSession()
api_call = slack_client.api_call("users.list")
if api_call.get('ok'):
# retrieve all users so we can find our bot
users = api_call.get('members')
for user in users:
if user['name'] == 'slackbot':
continue
if user['deleted']:
continue
if not user['is_bot']:
_sync_user(user, session)
else:
logger.warning('Error', api_call)
session.commit()
def delete_user(user):
"""Sync user with slack."""
session = DBSession()
db_user = session.query(User).filter(User.name == user).first()
if not db_user:
logger.warning('User {0} does not exist'.format(user))
exit(0)
balance = db_user.balance
logger.info('test {0} {1}'.format(user, balance))
if balance == 0.0:
session.delete(db_user)
session.commit()
logger.info('User {0} deleted !'.format(user))
else:
logger.warning('User {0} has a balance of {1}'.format(user, balance))
logger.warning('Must be reseted to 0 :)')
exit(1)
def achat(self, data, user_data, amount, description):
session = DBSession()
today = datetime.date.today()
user = self.get_db_user(session, user_data['name'])
expense = Expense(user_id=user.id,
amount=amount,
description=description)
session.add(expense)
session.commit()
send_info(data['channel'],
text='Achat enregistré !',
markdown=True,
thread=data["ts"])
def quimange(self, data, user_data):
outputs = []
session = DBSession()
day = self.get_db_day(session)
if not day:
day = Day(date=datetime.date.today())
session.add(day)
session.commit()
presences = session.query(Presence).filter(
Presence.day_id == day.id).all()
if not presences:
outputs.append("Personne d'inscrit aujourd'hui, sniff :(")
else:
total = 0
for presence in presences:
user = presence.user
total += presence.meals
guest = presence.meals - 1
if guest > 0:
outputs.append("{0} avec {1} invités".format(
user.name, guest))
else:
outputs.append(user.name)
if presence.cook == 1:
outputs.append("C'est {0} qui cuisinera".format(user.name))
if total > 1:
outputs.insert(0,
"Ce midi, {0} personnes mangent:".format(total))
else:
outputs.insert(
0, "Ce midi, {0} personne mange, bravo à elle !!".format(
total))
send_info(data['channel'], text='\n'.join(outputs), markdown=True)
def cancelcuisiner(self, data, user_data):
session = DBSession()
day = self.get_db_day(session)
if not day:
day = Day(date=datetime.date.today())
session.add(day)
session.commit()
outputs = []
user = self.get_db_user(session, user_data['name'])
if not user:
send_error(data['channel'],
'Erreur zjaifnazgoizangoiazg',
thread=data["ts"])
presence = session.query(Presence).filter(
Presence.user_id == user.id, Presence.day_id == day.id).first()
if not presence:
outputs.append(
"Tu n'étais pas inscrit, pas de soucis <@{0}> !".format(
user_data['id']))
else:
presence.cook = 0
session.add(presence)
session.commit()
outputs.append("J'ai supprimé le fait que tu avais cuisiné")
send_info(data['channel'],
text='\n'.join(outputs),
markdown=True,
ephemeral_user=user_data["id"])
data['text'] = '!QuiMange'
self.quimange(data, user_data)
def cuisiner(self, data, user_data):
session = DBSession()
day = self.get_db_day(session)
if not day:
day = Day(date=datetime.date.today())
session.add(day)
session.commit()
outputs = []
user = self.get_db_user(session, user_data['name'])
if not user:
send_error(data['channel'],
'Erreur il faut ```python manage.py sync``` d\'abort !',
thread=data["ts"])
presence = session.query(Presence).filter(
Presence.user_id == user.id, Presence.day_id == day.id).first()
if not presence:
presence = Presence(user_id=user.id, day_id=day.id, cook=1)
session.add(presence)
session.commit()
outputs.append(
"J'ai pris en compte le fait que tu cuisinais aujourd'hui")
else:
presence.cook = 1
session.add(presence)
session.commit()
outputs.append(
"J'ai pris en compte le fait que tu cuisinais aujourd'hui")
send_info(data['channel'],
text='\n'.join(outputs),
markdown=True,
ephemeral_user=user_data["id"])
data['text'] = '!QuiMange'
self.quimange(data, user_data)
def mybalance(self, data, user_data, history, for_user):
today = datetime.date.today()
session = DBSession()
if history < 1:
send_error(data['channel'],
"Non non non, history doit être > 1 <@{0}> ;)".format(
user_data['id']),
markdown=True)
return
if for_user:
user = self.get_db_user(session, for_user)
else:
user = self.get_db_user(session, user_data['name'])
if not user:
send_error(data['channel'], 'Erreur zjaifnazgoizangoiazg')
delta = datetime.timedelta(days=history)
start_date = today - delta
expenses = session.query(Expense).filter(
Expense.user_id == user.id, Expense.date >= start_date).all()
outputs = []
if not expenses:
outputs.append(
"Pas d'achat dans les {0} derniers jours pour {1}".format(
history, user.name))
else:
if for_user:
outputs.append(
"Liste des achats depuis {0} jours par {1} :".format(
history, user.name))
else:
outputs.append(
"Liste de tes achats depuis {0} jours :".format(history))
for expense in expenses:
text = "- Le {0} : {1} €".format(expense.date, expense.amount)
if expense.description:
text += " pour {0}".format(expense.description)
outputs.append(text)
presences = session.query(Presence).filter(
Presence.user_id == user.id).join(Day).filter(
Day.date >= start_date).all()
if not presences:
outputs.append(
"Pas de repas compatibilisé dans les {0} derniers jours pour {1}"
.format(history, user.name))
else:
if for_user:
outputs.append(
"Liste des repas depuis {0} jours de {1} :".format(
history, user.name))
else:
outputs.append(
"Liste de tes repas depuis {0} jours :".format(history))
for presence in presences:
text = "- Le {0} : {1} repas à {2} €".format(
presence.day.date, presence.meals, presence.day.price)
outputs.append(text)
if for_user:
outputs.append(
"Du coup, en tout, voici la balance de {0} : {1} €".format(
user.name, user.balance))
else:
outputs.append("Du coup, en tout, voici ta balance : {0} €".format(
user.balance))
send_info(data['channel'],
text='\n'.join(outputs),
markdown=True,
thread=data["ts"])
def cancelmanger(self, data, user_data, for_user):
session = DBSession()
day = self.get_db_day(session)
if not day:
day = Day(date=datetime.date.today())
session.add(day)
session.commit()
outputs = []
if for_user:
user = self.get_db_user(session, for_user)
else:
user = self.get_db_user(session, user_data['name'])
if not user:
send_error(data['channel'],
'Erreur zjaifnazgoizangoiazg',
thread=data["ts"])
presence = session.query(Presence).filter(
Presence.user_id == user.id, Presence.day_id == day.id).first()
if not presence:
if for_user:
outputs.append(
"@{0} n'étais pas inscrit, pas de soucis <@{1}> !".format(
user.name, user_data['id']))
else:
outputs.append(
"Tu n'étais pas inscrit, pas de soucis <@{0}> !".format(
user_data['id']))
else:
session.query(Presence).filter_by(id=presence.id).delete()
session.commit()
if for_user:
outputs.append("J'ai supprimé l'inscription de @{0}.".format(
user.name))
else:
outputs.append(
"J'ai supprimé ta demande <@{0}>, dommage ça allait être trop bon !"
.format(user_data['id']))
outputs.append("Si besoin tu !Manger à nouveau :)")
send_info(data['channel'],
text='\n'.join(outputs),
markdown=True,
ephemeral_user=user_data["id"])
data['text'] = '!QuiMange'
self.quimange(data, user_data)
def manger(self, data, user_data, guest, for_user):
session = DBSession()
if guest < 0:
send_error(
data['channel'],
"Touche à ton cul avec ton nombre négatif <@{0}> ;)".format(
user_data['id']),
markdown=True,
thread=data["ts"])
return
day = self.get_db_day(session)
if not day:
day = Day(date=datetime.date.today())
session.add(day)
session.commit()
outputs = []
if for_user:
user = self.get_db_user(session, for_user)
from_user = self.get_db_user(session, user_data['name'])
else:
user = self.get_db_user(session, user_data['name'])
if not user:
send_error(data['channel'],
'Erreur il faut ```python manage.py sync``` d\'abort !',
thread=data["ts"])
presence = session.query(Presence).filter(
Presence.user_id == user.id, Presence.day_id == day.id).first()
if not presence:
presence = Presence(user_id=user.id,
day_id=day.id,
meals=guest + 1)
session.add(presence)
session.commit()
if for_user:
outputs.append(
"Demande enregistrée par <@{0}> pour <@{1}>".format(
from_user.slackid, user.slackid))
if guest > 0:
outputs.append("Je compterai {0} part(s)".format(guest +
1))
else:
outputs.append("J'ai pris en compte ta demande <@{0}>".format(
user_data['id']))
if guest > 0:
outputs.append(
"Je te compterai {0} part ce jour".format(guest + 1))
else:
if presence.meals != guest + 1:
presence.meals = guest + 1
session.add(presence)
session.commit()
if for_user:
if guest > 0:
outputs.append(
"Demande modifiée par <@{0}> pour <@{1}>, {2} invité(s) rajoutés"
.format(from_user.slackid, user.slackid, guest))
else:
outputs.append(
"Demande modifiée par <@{0}> pour <@{1}>, les invités ont été supprimés"
.format(from_user.slackid, user.slackid, guest))
else:
if guest > 0:
outputs.append(
"J'ai modifié ta demande, et rajouté {0} invité(s) <@{1}>"
.format(guest, user.slackid))
else:
outputs.append(
"J'ai modifié ta demande, et enlevé les invités <@{1}>"
.format(guest, user.slackid))
else:
if for_user:
outputs.append(
"Rien n'a changé, <@{0}> était déjà inscrit <@{1}> :)".
format(user.slackid, from_user.slackid))
else:
outputs.append(
"Rien n'a changé, tu étais déjà inscrit <@{0}> :)".
format(user.slackid))
outputs.append(
"Si besoin tu peux !CancelManger ou !Manger avec des invités :)")
if for_user:
send_info(data['channel'],
text='\n'.join(outputs),
markdown=True,
ephemeral_user=user.slackid)
send_info(data['channel'],
text='\n'.join(outputs),
markdown=True,
ephemeral_user=from_user.slackid)
else:
send_info(data['channel'],
text='\n'.join(outputs),
markdown=True,
ephemeral_user=user_data["id"])
data['text'] = '!QuiMange'
self.quimange(data, user_data)
def balance(self, data, user_data):
session = DBSession()
users = (session.query(User).filter(User.status == 1).all())
outputs = []
outputs.append('Voici l\'état des compte chez Louise :')
real_users = []
for user in users:
real_users.append({'name': user.name, 'balance': user.balance})
total_balance = 0.0
for user in sorted(real_users, key=lambda x: x['balance']):
user_balance = float(user['balance'])
total_balance += user_balance
if abs(user_balance) > 1:
outputs.append('{0} : balance à {1:.2f}'.format(
user['name'], user_balance))
outputs.append('\nTotal balance : {0:.2f}'.format(total_balance))
send_info(data['channel'], text='\n'.join(outputs))
query = session.query(User.name, User.status, User.id.label('id'), func.sum(Presence.cook).label('sum_cook'))\
.filter(User.status == 1)\
.join(Presence, User.id == Presence.user_id).filter(Presence.cook.isnot(None)).group_by(User.name)
users = query.all()
outputs = []
#outputs.append('Qui a cuisiné et pour combien de personnes ?')
id_user = 0
outputs.append('```')
outputs.append(
'| Nom | Repas cuisiné | Repas mangé| RC / RM |')
outputs.append(
'|-------------|---------------|------------|---------|')
for user in users:
# Skip old user
if not user.status:
continue
days = session.query(
Presence.day_id).filter(Presence.cook == 1).filter(
Presence.user_id == user.id).group_by(
Presence.day_id).subquery() # days where he cooked
cooked = session.query(
Presence.user_id,
func.sum(Presence.meals).label('sum_cooked')).filter(
Presence.day_id.in_(days)).first(
) #sum_cooked Number of person that he cooked for
if cooked is None:
cooked = 0
else:
cooked = str(
cooked[1]
) #cooked arrive avec le format suivant [(10, 2)] , donc [(user_id, sum_cooked)]
eat = session.query(func.count(Presence.cook)).filter(
Presence.cook.isnot(None)).filter(
Presence.user_id ==
user.id).first() # Number of days he eated
try:
rapport = round(int(cooked) / int(eat[0]), 2)
except ValueError:
rapport = 0
outputs.append('|{:13}|{:15}|{:12}|{:9}|'.format(
user.name, cooked, eat[0], rapport))
outputs.append('```')
send_info(data['channel'], text='\n'.join(outputs), markdown=True)
