|
|
from datetime import datetime, timedelta, date
|
|
|
from datetime import time as dtime
|
|
|
|
|
|
from django.core.management import BaseCommand
|
|
|
from django.db import transaction
|
|
|
from django.utils import timezone
|
|
|
|
|
|
from boards.models import Post
|
|
|
from boards.models.post.manager import NO_IP
|
|
|
|
|
|
__author__ = 'neko259'
|
|
|
|
|
|
|
|
|
class Command(BaseCommand):
|
|
|
help = 'Removes user and IP data from posts'
|
|
|
|
|
|
def add_arguments(self, parser):
|
|
|
parser.add_argument('--days-before', type=int,
|
|
|
help='Clean posts only before the number of days before today.')
|
|
|
|
|
|
@transaction.atomic
|
|
|
def handle(self, *args, **options):
|
|
|
days_before = options.get('days_before')
|
|
|
|
|
|
if days_before:
|
|
|
day_end = date.today() - timedelta(1)
|
|
|
|
|
|
day_time_end = timezone.make_aware(datetime.combine(
|
|
|
day_end, dtime()), timezone.get_current_timezone())
|
|
|
posts = Post.objects.filter(pub_time__lte=day_time_end)
|
|
|
else:
|
|
|
posts = Post.objects.all()
|
|
|
posts.update(poster_ip=NO_IP)
|
|
|
|
|
|
|
