class Gold_Extractor(db.DynamicDocument):
name = db.StringField()
model_file = db.FileField()
model_meta_file = db.FileField()
vocabulary = db.StringField()
class Snapshots(db.Document):
url = db.StringField(max_length=1000, required=True)
timestamp = db.DateTimeField(required=False, default=datetime.datetime.now)
response = db.EmbeddedDocumentField(Responses)
sha256 = db.StringField(max_length=256, required=True)
screenshot = db.FileField()
meta = {
'ordering': ['-timestamp'],
}
class Files(db.Document):
filedata = db.FileField()
filename = db.StringField()
library = db.StringField()
tag = db.ListField()
library_md5 = db.StringField()
author = db.EmailField()
status = db.StringField()
text = db.StringField()
html = db.StringField()
# 全文搜索
def search(self, library, option, search_name):
if search_name:
# files = None
if option == u'全文搜索':
a = library
if library == u'全部':
files = Files.objects(
__raw__={
"$or": [{
'html': re.compile(search_name)
}, {
'filename': re.compile(search_name)
}]
})
files_list, count, match_count = fulltext_search1(
files, search_name)
dict1 = {
'files_list': files_list,
'count': count,
'match_count': match_count
}
return dict1
else:
files = Files.objects(
__raw__={
'library':
library,
"$or": [{
'html': re.compile(search_name)
}, {
'filename': re.compile(search_name)
}]
})
files_list, count, match_count = fulltext_search1(
files, search_name)
dict1 = {
'files_list': files_list,
'count': count,
'match_count': match_count
}
return dict1
elif option == u'文件搜索':
if library == u'全部':
files = Files.objects(
__raw__={'filename': re.compile(search_name)})
files_list, count, match_count = filename_search(files)
dict1 = {
'files_list': files_list,
'count': count,
'match_count': match_count
}
return dict1
else:
files = Files.objects(__raw__={
'library': library,
'filename': re.compile(search_name)
})
files_list, count, match_count = filename_search(files)
dict1 = {
'files_list': files_list,
'count': count,
'match_count': match_count
}
return dict1
class TaskResults(db.Document):
beaconId = db.StringField(max_length=150, required=True)
taskId = db.StringField(max_length=10, required=True)
taskEndDate = db.DateTimeField()
taskOutput = db.FileField()
class MasterFile(db.EmbeddedDocument): file_composition = db.ListField(db.ObjectIdField()) file = db.FileField()
class File(db.EmbeddedDocument): file_ = db.FileField()