def get_user(self, message):
connect()
assert isinstance(message, Message)
user = message.sender
# user_id = user.user.split("_")[1]
# hipchat_user = self.get_hipchat_user(user_id)
# available on hipchat_user
user = User.get_or_create(user.hipchat_id, user.name, user.nick)
return user
def summary_view(self, message):
connect()
result = []
for user in User.objects():
updates = StatusUpdateUserAggregated.objects(user_id=user.user_id).limit(5)
if not len(updates):
continue
template = rendered_template("show_updates.html", {"user": user, "updates": updates})
self.reply(message, template, html=True)
def get_user(self, message):
connect()
assert isinstance(message, Message)
user = message.sender
# user_id = user.user.split("_")[1]
# hipchat_user = self.get_hipchat_user(user_id)
# available on hipchat_user
user = User.get_or_create(user.hipchat_id,
user.name,
user.nick)
return user
def summary_view(self, message):
connect()
result = []
for user in User.objects():
updates = StatusUpdateUserAggregated.objects(user_id=user.user_id).limit(5)
if not len(updates):
continue
template = rendered_template("show_updates.html",
{"user":user, "updates":updates})
self.reply(message, template, html=True)
def update_status(self, message, project_name, status):
connect()
# insert into postgres
user = self.get_user(message)
try:
project = Project.get(user_id=user.user_id, name=project_name)
except Project.DoesNotExist:
self.reply(message, "Sorry, project %s not found." % project_name)
return
status = StatusUpdate.create(project, status)
self.reply(message, "Status updated." + str(status))
def update_status(self, message, project_name, status):
connect()
# insert into postgres
user = self.get_user(message)
try:
project = Project.get(user_id=user.user_id, name=project_name)
except Project.DoesNotExist:
self.reply(message, "Sorry, project %s not found." % project_name)
return
status = StatusUpdate.create(project, status)
self.reply(message, "Status updated." + str(status))
def lsproject(self, message, nick):
# lists all projects, current user is assumed
connect()
if nick:
nick = nick.strip()
user = User.get_by_nick(nick)
else:
user = self.get_user(message)
if not user:
self.reply(message, "Could not find user %s" % nick)
return
projects = Project.objects(user_id=user.user_id)
template = rendered_template("lsproject.html", {"user":user, "projects":projects})
self.reply(message, template, html=True)
def lsproject(self, message, nick):
# lists all projects, current user is assumed
connect()
if nick:
nick = nick.strip()
user = User.get_by_nick(nick)
else:
user = self.get_user(message)
if not user:
self.reply(message, "Could not find user %s" % nick)
return
projects = Project.objects(user_id=user.user_id)
template = rendered_template("lsproject.html", {"user":user, "projects":projects})
self.reply(message, template, html=True)
def show_updates(self, message, nick):
# shows updates for a single user
connect()
try:
if not nick:
user = self.get_user(message)
else:
user = User.get_by_nick(nick)
except User.DoesNotExist:
self.reply(message, "%s does not exist yet." % nick)
return
updates = StatusUpdateUserAggregated.objects(user_id=user.user_id).limit(10)
template = rendered_template("show_updates.html", {"user": user, "updates": updates})
self.reply(message, template, html=True)
def show_updates(self, message, nick):
# shows updates for a single user
connect()
try:
if not nick:
user = self.get_user(message)
else:
user = User.get_by_nick(nick)
except User.DoesNotExist:
self.reply(message, "%s does not exist yet." % nick)
return
updates = StatusUpdateUserAggregated.objects(user_id=user.user_id).limit(10)
template = rendered_template("show_updates.html",
{"user":user, "updates":updates})
self.reply(message, template, html=True)
#!/usr/bin/env python
import sys
import nose
sys.path.append('')
from meatbot.status import connect
connect()
nose.main()
#!/usr/bin/env python
import sys
import nose
sys.path.append('')
from meatbot.status import connect
connect()
nose.main()
