Преглед изворни кода

sandbox doesn't provide auto disable log (#12388)

tags/0.15.0
Jyong пре 10 месеци
родитељ
комит
9c317b64c3
No account linked to committer's email address
2 измењених фајлова са 6 додато и 17 уклоњено
  1. 0
    17
      api/schedule/clean_unused_datasets_task.py
  2. 6
    0
      api/services/dataset_service.py

+ 0
- 17
api/schedule/clean_unused_datasets_task.py Прегледај датотеку

else: else:
plan = plan_cache.decode() plan = plan_cache.decode()
if plan == "sandbox": if plan == "sandbox":
# add auto disable log
documents = (
db.session.query(Document)
.filter(
Document.dataset_id == dataset.id,
Document.enabled == True,
Document.archived == False,
)
.all()
)
for document in documents:
dataset_auto_disable_log = DatasetAutoDisableLog(
tenant_id=dataset.tenant_id,
dataset_id=dataset.id,
document_id=document.id,
)
db.session.add(dataset_auto_disable_log)
# remove index # remove index
index_processor = IndexProcessorFactory(dataset.doc_form).init_index_processor() index_processor = IndexProcessorFactory(dataset.doc_form).init_index_processor()
index_processor.clean(dataset, None) index_processor.clean(dataset, None)

+ 6
- 0
api/services/dataset_service.py Прегледај датотеку



@staticmethod @staticmethod
def get_dataset_auto_disable_logs(dataset_id: str) -> dict: def get_dataset_auto_disable_logs(dataset_id: str) -> dict:
features = FeatureService.get_features(current_user.current_tenant_id)
if not features.billing.enabled or features.billing.subscription.plan == "sandbox":
return {
"document_ids": [],
"count": 0,
}
# get recent 30 days auto disable logs # get recent 30 days auto disable logs
start_date = datetime.datetime.now() - datetime.timedelta(days=30) start_date = datetime.datetime.now() - datetime.timedelta(days=30)
dataset_auto_disable_logs = DatasetAutoDisableLog.query.filter( dataset_auto_disable_logs = DatasetAutoDisableLog.query.filter(

Loading…
Откажи
Сачувај