Produits Entr'ouvert » Fargo

sudo -u fargo /usr/bin/fargo-manage tenant_command fargo-cleanup --all

from django.core.management.base import NoArgsCommand

from fargo.fargo.utils import cleanup

class Command(NoArgsCommand):

    help = 'Clean expired models of authentic2.'

    def handle_noargs(self, **options):

        cleanup()

import datetime

from django.utils.timezone import now

    def cleanup(self, n=None):

        '''Delete all orphaned documents'''

        n = n or now()

        # use a window of 60 seconds to be sure this document will never be used

        qs = self.filter(creation_date__lt=n - datetime.timedelta(seconds=60))

        qs = qs.filter(user_documents__isnull=True,

                       oauth2_tempfiles__isnull=True)

            document.content.delete(False)

# -*- coding: utf-8 -*-

# Generated by Django 1.11.11 on 2018-03-23 15:30

from __future__ import unicode_literals

from django.db import migrations, models

class Migration(migrations.Migration):

    dependencies = [

        ('fargo', '0014_auto_20171016_0854'),

    ]

    operations = [

        migrations.AddField(

            model_name='document',

            name='creation_date',

            field=models.DateTimeField(auto_now_add=True),

        ),

    ]

    creation_date = models.DateTimeField(auto_now_add=True)

from django.db import models

def cleanup_model(model, n=None):

    manager = getattr(model, 'objects', None)

    if hasattr(manager, 'cleanup'):

        manager.cleanup(n=n)

def cleanup(n=None):

    for app in models.get_apps():

        for model in models.get_models(app):

            cleanup_model(model, n=n)

                ('document', models.ForeignKey(to='fargo.Document', related_name='oauth2_tempfiles')),

import datetime

from django.conf import settings

from django.db.models.query import QuerySet

from django.utils.timezone import now

class CleanupQuerySet(QuerySet):

    def cleanup(self, n=None):

        n = n or now()

        threshold = n - datetime.timedelta(seconds=2 * self.model.get_lifetime())

        self.filter(creation_date__lt=threshold).delete()

    objects = CleanupQuerySet.as_manager()

    @classmethod

    def get_lifetime(cls):

        return max(

            settings.FARGO_CODE_LIFETIME,

            settings.FARGO_ACCESS_TOKEN_LIFETIME)

    document = models.ForeignKey(Document, related_name='oauth2_tempfiles')

    objects = CleanupQuerySet.as_manager()

    @classmethod

    def get_lifetime(cls):

        return settings.FARGO_OAUTH2_TEMPFILE_LIFETIME

FARGO_OAUTH2_TEMPFILE_LIFETIME = 86400

import datetime

from django.core.management import call_command

from django.contrib.auth.models import User

from fargo.fargo.models import UserDocument, Document

from fargo.oauth2.models import OAuth2TempFile, OAuth2Client

from django.core.files.base import ContentFile

def test_cleanup(freezer, john_doe):

    start = freezer()

    client = OAuth2Client.objects.create(client_name='c', redirect_uris='')

    foo = Document.objects.create(content=ContentFile('foo', name='foo.txt'))

    bar = Document.objects.create(content=ContentFile('bar', name='bar.txt'))

    UserDocument.objects.create(user=john_doe,

                                document=foo,

                                filename='foo.txt',

                                title='',

                                description='')

    OAuth2TempFile.objects.create(document=bar, client=client, filename='bar.txt')

    call_command('fargo-cleanup')

    assert UserDocument.objects.all().count()

    assert OAuth2TempFile.objects.all().count()

    assert Document.objects.all().count() == 2

    User.objects.all().delete()

    assert not UserDocument.objects.all().count()

    assert Document.objects.all().count() == 2

    call_command('fargo-cleanup')

    assert Document.objects.all().count() == 2

    freezer.move_to(start + datetime.timedelta(seconds=120))

    call_command('fargo-cleanup')

    assert Document.objects.all().count() == 1

    freezer.move_to(start + datetime.timedelta(days=3))

    call_command('fargo-cleanup')

    assert not OAuth2TempFile.objects.count()

    assert Document.objects.count()

    call_command('fargo-cleanup')

    assert not Document.objects.count()

        sqlite: DB_ENGINE=sqlite3

        pg: DB_ENGINE=postgresql_psycopg2

        DJANGO_SETTINGS_MODULE=fargo.settings

        FARGO_SETTINGS_FILE=tests/settings.py

        coverage: COVERAGE=--cov=fargo --cov-report xml

        dj18: django>=1.8,<1.9

        dj18: django-tables2<1.1

        dj111: django>=1.11,<1.12

        dj111: django-tables2>=1.5

        pytest>=3.3.0

        pytest-cov

        pytest-random

        pytest-mock

        pytest-django

        pytest-freezegun

        django-webtest

        WebTest

        djangorestframework>=3.3,<3.4

        py.test {env:COVERAGE:} {posargs:--random --junit-xml=junit-{envname}.xml tests}

        coverage: mv coverage.xml coverage-{envname}.xml