import DBHelper
import argparse
import sys
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='input startPage, endPage')
parser.add_argument('--startPage', type=int, help='input start page')
parser.add_argument('--endPage', type=int, help='input start page')
args = parser.parse_args()
for i in DBHelper.GetDB(args.startPage, args.endPage):
if 'abuse' in i.keys():
continue
print(i['text'])
while True:
stdin = input("good 0, bad 1, skip 2, exit 3: ")
if stdin == "3":
exit(0)
elif stdin == "1":
data = {'abuse': '1'}
DBHelper.UpdateDB(i['_id']['$oid'], data)
break
elif stdin == "0":
data = {'abuse': '0'}
DBHelper.UpdateDB(i['_id']['$oid'], data)
break
elif stdin == "2":
print("skip")
break
import DBHelper
if __name__ == "__main__":
f = open("NotEvil.txt", 'w',encoding="utf-8")
f2 = open("Evil.txt", 'w',encoding="utf-8")
f3 = open("NotAbuse.txt", 'w',encoding="utf-8")
f4 = open("Abuse.txt", 'w',encoding="utf-8")
for i in DBHelper.GetDB(None):
if 'evil' not in i.keys():
pass
elif i['evil']=="0":
#f.write(i["_id"]["$oid"]+" "+i['text']+'\n')
f.write(i['text']+'\n')
elif i['evil']=="1":
#f2.write(i["_id"]["$oid"]+" "+i['text']+'\n')
f2.write(i['text']+'\n')
if 'abuse' not in i.keys():
pass
elif i['abuse']=="0":
f3.write(i['text']+'\n')
elif i['abuse']=="1":
f4.write(i['text']+'\n')
f.close()
f2.close()
f3.close()
f4.close()