You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
2.1 KiB
62 lines
2.1 KiB
from _arango import arango |
|
|
|
def truncate(): |
|
arango.db.collection("other_persons").truncate() |
|
arango.db.collection("all_relations").truncate() |
|
|
|
|
|
def clear_info_persons(): |
|
persons = list(arango.db.collection("persons").all()) |
|
for person in persons: |
|
if 'other' in person: |
|
if person['other']: |
|
arango.db.collection('persons').delete(person) |
|
continue |
|
person['info'] = [] |
|
person['mentioned_in_interrogation'] = [] |
|
arango.db.collection('persons').update(person, merge=False) |
|
|
|
def clear_changer_interrogations(): |
|
interrogations = list(arango.db.collection("interrogations").all()) |
|
for interrogation in interrogations: |
|
interrogation['mentioned_persons'] = [] |
|
arango.db.collection('interrogations').update(interrogation, merge=False) |
|
|
|
|
|
def clean_mentioned_as(): |
|
persons = list(arango.db.collection("persons").all()) |
|
for person in persons: |
|
if 'mentioned_as' in person: |
|
mentioned_as = [] |
|
for i in person['mentioned_as']: |
|
if i not in mentioned_as: |
|
mentioned_as.append(i) |
|
person['mentioned_as'] = mentioned_as |
|
arango.db.collection('persons').update(person, merge=False) |
|
|
|
db = arango.db |
|
|
|
cursor = db.aql.execute('for doc in rumors return doc') |
|
rumors = list(cursor) |
|
|
|
for rumor in rumors: |
|
rumor['class'] = rumor['class'].replace('.', '').strip().lower() |
|
db.collection('rumors').update(rumor, merge=False) |
|
|
|
|
|
|
|
|
|
# persons = list(arango.db.collection("persons").all()) |
|
|
|
# for person in persons: |
|
# for interrogation in person['interrogations']: |
|
|
|
# arango.db.collection('interrogations').update({'_key': interrogation, 'person_id': person['_id']}, ) |
|
# print(f"Updated {interrogation} with person_id {person['_id']}") |
|
|
|
|
|
# interrogations = list(arango.db.collection("interrogations").all()) |
|
|
|
# for interrogation in interrogations: |
|
# interrogation['person_id'] = 'persons/' + interrogation['person'].replace('persons_', '') |
|
# arango.db.collection('interrogations').update(interrogation, merge=False) |