mirror of https://github.com/zulip/zulip.git
37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
from __future__ import absolute_import
|
|
|
|
from django.core.management.base import BaseCommand
|
|
|
|
from zephyr.lib.actions import update_message_flags
|
|
from zephyr.models import UserProfile, Message, get_user_profile_by_email
|
|
|
|
class Command(BaseCommand):
|
|
help = """Bankrupt one or many users.
|
|
|
|
Usage: python manage.py bankrupt_users <list of email addresses>"""
|
|
|
|
def handle(self, *args, **options):
|
|
if len(args) < 1:
|
|
print "Please provide at least one e-mail address."
|
|
exit(1)
|
|
|
|
for email in args:
|
|
try:
|
|
user_profile = get_user_profile_by_email(email)
|
|
except UserProfile.DoesNotExist:
|
|
print "e-mail %s doesn't exist in the system, skipping" % (email,)
|
|
continue
|
|
|
|
update_message_flags(user_profile, "add", "read", None, True)
|
|
|
|
messages = Message.objects.filter(
|
|
usermessage__user_profile=user_profile).order_by('-id')[:1]
|
|
if messages:
|
|
old_pointer = user_profile.pointer
|
|
new_pointer = messages[0].id
|
|
user_profile.pointer = new_pointer
|
|
user_profile.save(update_fields=["pointer"])
|
|
print "%s: %d => %d" % (email, old_pointer, new_pointer)
|
|
else:
|
|
print "%s has no messages, can't bankrupt!" % (email,)
|