def handle(self, *args, **options):
# type: (*Any, **Any) -> None
if not options["flag"] or not options["op"] or not options["email"]:
print("Please specify an operation, a flag and an email")
exit(1)
op = options['op']
flag = getattr(UserMessage.flags, options['flag'])
all_until = options['all_until']
email = options['email']
realm = self.get_realm(options)
user_profile = self.get_user(email, realm)
if all_until:
filt = models.Q(id__lte=all_until)
else:
filt = models.Q(message__id__in=[
mid.strip() for mid in sys.stdin.read().split(',')
])
mids = [
m.id for m in UserMessage.objects.filter(
filt, user_profile=user_profile).order_by('-id')
]
if options["for_real"]:
sys.stdin.close()
sys.stdout.close()
sys.stderr.close()
def do_update(batch):
# type: (Iterable[int]) -> None
msgs = UserMessage.objects.filter(id__in=batch)
if op == 'add':
msgs.update(flags=models.F('flags').bitor(flag))
elif op == 'remove':
msgs.update(flags=models.F('flags').bitand(~flag))
if not options["for_real"]:
logging.info("Updating %s by %s %s" % (mids, op, flag))
logging.info(
"Dry run completed. Run with --for-real to change message flags."
)
exit(1)
utils.run_in_batches(mids, 400, do_update, sleep_time=3)
exit(0)
def handle(self, *args: Any, **options: Any) -> None:
if not options["flag"] or not options["op"] or not options["email"]:
raise CommandError(
"Please specify an operation, a flag and an email")
op = options["op"]
flag = getattr(UserMessage.flags, options["flag"])
all_until = options["all_until"]
email = options["email"]
realm = self.get_realm(options)
user_profile = self.get_user(email, realm)
if all_until:
filt = models.Q(id__lte=all_until)
else:
filt = models.Q(message_id__in=[
mid.strip() for mid in sys.stdin.read().split(",")
])
mids = [
m.id for m in UserMessage.objects.filter(
filt, user_profile=user_profile).order_by("-id")
]
if options["for_real"]:
sys.stdin.close()
sys.stdout.close()
sys.stderr.close()
def do_update(batch: Collection[int]) -> None:
msgs = UserMessage.objects.filter(id__in=batch)
if op == "add":
msgs.update(flags=models.F("flags").bitor(flag))
elif op == "remove":
msgs.update(flags=models.F("flags").bitand(~flag))
if not options["for_real"]:
logging.info("Updating %s by %s %s", mids, op, flag)
logging.info(
"Dry run completed. Run with --for-real to change message flags."
)
raise CommandError
utils.run_in_batches(mids, 400, do_update, sleep_time=3)
exit(0)
def handle(self, *args, **options):
# type: (*Any, **Any) -> None
if not options["flag"] or not options["op"] or not options["email"]:
print("Please specify an operation, a flag and an email")
exit(1)
op = options['op']
flag = getattr(UserMessage.flags, options['flag'])
all_until = options['all_until']
email = options['email']
user_profile = get_user_profile_by_email(email)
if all_until:
filt = models.Q(id__lte=all_until)
else:
filt = models.Q(message__id__in=[mid.strip() for mid in sys.stdin.read().split(',')])
mids = [m.id for m in
UserMessage.objects.filter(filt, user_profile=user_profile).order_by('-id')]
if options["for_real"]:
sys.stdin.close()
sys.stdout.close()
sys.stderr.close()
def do_update(batch):
# type: (Iterable[int]) -> None
msgs = UserMessage.objects.filter(id__in=batch)
if op == 'add':
msgs.update(flags=models.F('flags').bitor(flag))
elif op == 'remove':
msgs.update(flags=models.F('flags').bitand(~flag))
if not options["for_real"]:
logging.info("Updating %s by %s %s" % (mids, op, flag))
logging.info("Dry run completed. Run with --for-real to change message flags.")
exit(1)
utils.run_in_batches(mids, 400, do_update, sleep_time=3)
exit(0)