diff --git a/src/api.py b/src/api.py index 0810b51..c5a6f63 100644 --- a/src/api.py +++ b/src/api.py @@ -5,7 +5,7 @@ django.setup() from ctirs.core.mongo.documents_stix import StixFiles -from ctirs.core.mongo.documents import TaxiiServers, InformationSources, Vias, Communities +from ctirs.core.mongo.documents import TaxiiServers, Vias, Communities import yaml import datetime import dateutil.tz @@ -28,8 +28,6 @@ from stix.extensions.marking.ais import AISMarkingStructure - - class StipTaxiiServerAPI(OpenTAXIIPersistenceAPI): STIP_SNS_USER_NAME_PREFIX = 'User Name: ' @@ -152,7 +150,7 @@ def get_stix_files_from_mongo(self, collection_name, start_time, end_time): ts = TaxiiServers.objects.get(collection_name=collection_name) # 条件絞り込み - QQ = Q(information_source__in=ts.information_sources) + QQ = Q(input_community__in=ts.communities) QQ = QQ & Q (version__startswith='1.') if start_time is not None and end_time is not None: QQ = QQ & Q(produced__gt=start_time)