Files
zulip/zerver/management/commands/bankrupt_users.py
Tim Abbott e111a2f9a5 [manual] Rename Django app from zephyr to zerver.
This needs to be deployed to both staging and prod at the same
off-peak time (and the schema migration run).

At the time it is deployed, we need to make a few changes directly in
the database:

(1) UPDATE django_content_type set app_label='zerver' where app_label='zephyr';
(2) UPDATE south_migrationhistory set app_name='zerver' where app_name='zephyr';

(imported from commit eb3fd719571740189514ef0b884738cb30df1320)
2013-08-06 07:39:36 -04:00

37 lines
1.3 KiB
Python

from __future__ import absolute_import
from django.core.management.base import BaseCommand
from zerver.lib.actions import update_message_flags
from zerver.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,)