mirror of https://github.com/zulip/zulip.git
51 lines
2.2 KiB
Python
51 lines
2.2 KiB
Python
|
from django.core.management.base import BaseCommand
|
||
|
|
from zephyr.models import UserProfile, Subscription, Recipient, Message, Stream
|
||
|
|
from django.db.models import Q
|
||
|
|
|
||
|
|
import datetime
|
||
|
|
import pytz
|
||
|
|
from optparse import make_option
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = """Delete all inactive tutorial stream subscriptions."""
|
||
|
|
|
||
|
|
option_list = BaseCommand.option_list + (
|
||
|
|
make_option('-f', '--for-real',
|
||
|
|
dest='for_real',
|
||
|
|
action='store_true',
|
||
|
|
default=False,
|
||
|
|
help="Actually deactive subscriptions. Default is a dry run."),
|
||
|
|
)
|
||
|
|
|
||
|
|
def has_sent_to(self, user_profile, recipient):
|
||
|
|
return Message.objects.filter(sender=user_profile, recipient=recipient).count() != 0
|
||
|
|
|
||
|
|
def handle(self, **options):
|
||
|
|
possible_tutorial_streams = Stream.objects.filter(Q(name__startswith='tutorial-'))
|
||
|
|
|
||
|
|
tutorial_bot = UserProfile.objects.get(user__email="humbug+tutorial@humbughq.com")
|
||
|
|
|
||
|
|
for stream in possible_tutorial_streams:
|
||
|
|
recipient = Recipient.objects.get(type=Recipient.STREAM, type_id=stream.id)
|
||
|
|
subscribers = Subscription.objects.filter(recipient=recipient, active=True)
|
||
|
|
if ((subscribers.count() == 1) and self.has_sent_to(tutorial_bot, recipient)):
|
||
|
|
# This is a tutorial stream.
|
||
|
|
most_recent_message = Message.objects.filter(
|
||
|
|
recipient=recipient).latest("pub_date")
|
||
|
|
# This cutoff must be more generous than the tutorial bot cutoff
|
||
|
|
# in the client code.
|
||
|
|
cutoff = datetime.datetime.now(tz=pytz.utc) - datetime.timedelta(hours=2)
|
||
|
|
|
||
|
|
if most_recent_message.pub_date < cutoff:
|
||
|
|
# The tutorial has expired, so delete the stream.
|
||
|
|
print stream.name, most_recent_message.pub_date
|
||
|
|
if options["for_real"]:
|
||
|
|
tutorial_user = subscribers[0]
|
||
|
|
tutorial_user.active = False
|
||
|
|
tutorial_user.save()
|
||
|
|
|
||
|
|
if options["for_real"]:
|
||
|
|
print "Subscriptions deactivated."
|
||
|
|
else:
|
||
|
|
print "This was a dry run. Pass -f to actually deactivate."
|