NewsBlur/utils/archive/bootstrap_intel.py

32 lines
1.4 KiB
Python
Raw Normal View History

import sys
from mongoengine.queryset import OperationError
from mongoengine.errors import ValidationError
from apps.analyzer.models import MClassifierFeed
from apps.analyzer.models import MClassifierAuthor
from apps.analyzer.models import MClassifierTag
from apps.analyzer.models import MClassifierTitle
for classifier_cls in [MClassifierFeed, MClassifierAuthor,
MClassifierTag, MClassifierTitle]:
2020-06-15 05:09:10 -04:00
print(" ================================================================= ")
print((" Now on %s " % classifier_cls.__name__))
print(" ================================================================= ")
classifiers = classifier_cls.objects.filter(social_user_id__exists=False)
count = classifiers.count()
2020-06-15 05:09:10 -04:00
print((" ---> Found %s classifiers" % count))
for i, classifier in enumerate(classifiers):
if i % 1000 == 0:
2020-06-15 05:09:10 -04:00
print((" ---> %s / %s" % (i, count)))
sys.stdout.flush()
classifier.social_user_id = 0
try:
classifier.save()
2020-06-15 05:09:10 -04:00
except OperationError as e:
print((" ***> Operation error on: %s" % e))
sys.stdout.flush()
# classifier.delete()
2020-06-15 05:09:10 -04:00
except ValidationError as e:
print((" ***> ValidationError error on: %s" % e))
print((" ***> Original classifier: %s" % classifier.__dict__))