NewsBlur/utils/archive/bootstrap_intel.py
2024-04-24 09:50:42 -04:00

33 lines
1.3 KiB
Python

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