mirror of
https://github.com/zulip/zulip.git
synced 2025-11-06 23:13:25 +00:00
761b7cd5fa
Otherwise one gets: AttributeError: 'module' object has no attribute 'time' when trying to use the time module from inside zephyr.lib. (imported from commit 645368672a3eff68320278dd480edeed56721fcc)
48 lines
1.7 KiB
Python
Executable File
48 lines
1.7 KiB
Python
Executable File
from optparse import make_option
|
|
from django.core.management.base import BaseCommand
|
|
from zephyr.models import Realm, UserProfile, Message, UserMessage
|
|
from zephyr.lib.timestamp import datetime_to_timestamp, timestamp_to_datetime
|
|
import simplejson
|
|
|
|
def dump():
|
|
pointers = []
|
|
for u in UserProfile.objects.select_related("user__email").all():
|
|
pointer = u.pointer
|
|
if pointer != -1:
|
|
pub_date = Message.objects.get(id=pointer).pub_date
|
|
pointers.append((u.user.email, datetime_to_timestamp(pub_date)))
|
|
else:
|
|
pointers.append((u.user.email, -1))
|
|
file("dumped-pointers", "w").write(simplejson.dumps(pointers) + "\n")
|
|
|
|
def restore(change):
|
|
for (email, timestamp) in simplejson.loads(file("dumped-pointers").read()):
|
|
try:
|
|
u = UserProfile.objects.get(user__email__iexact=email)
|
|
except UserProfile.DoesNotExist:
|
|
print "Skipping...", email
|
|
continue
|
|
if timestamp == -1:
|
|
pointer = -1
|
|
else:
|
|
try:
|
|
pointer = UserMessage.objects.filter(user_profile=u,
|
|
message__pub_date__gte=timestamp_to_datetime(timestamp)).order_by("message")[0].message_id
|
|
except IndexError:
|
|
print "Alert...", email, timestamp
|
|
continue
|
|
if change:
|
|
u.pointer = pointer
|
|
u.save()
|
|
|
|
class Command(BaseCommand):
|
|
option_list = BaseCommand.option_list + (
|
|
make_option('--restore', default=False, action='store_true'),
|
|
make_option('--dry-run', '-n', default=False, action='store_true'),)
|
|
|
|
def handle(self, *args, **options):
|
|
if options["restore"]:
|
|
restore(change=not options['dry_run'])
|
|
else:
|
|
dump()
|