summaryrefslogtreecommitdiffstats
path: root/askbot/management/commands/subscribe_everyone.py
blob: 12f7bed288de4473ad1ffc09ca0d6aeec763e253 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
from django.core.management.base import NoArgsCommand
from django.db import connection
from askbot.models import EmailFeedSetting, User

class Command(NoArgsCommand):
    def handle_noargs(self, **options):
        try:
            try:
                self.subscribe_everyone()
            except Exception, e:
                print e
        finally:
            connection.close()

    def subscribe_everyone(self):

        for user in User.objects.all():
            for feed_type in EmailFeedSetting.FEED_TYPES:
                try:
                    feed_setting = EmailFeedSetting.objects.get(
                                                subscriber=user,
                                                feed_type = feed_type[0]
                                            )
                except EmailFeedSetting.DoesNotExist:
                    feed_setting = EmailFeedSetting(
                                        subscriber=user,
                                        feed_type=feed_type[0]
                                    )
                feed_setting.frequency = 'w'
                feed_setting.reported_at = None
                feed_setting.save()