def __init__(self):
self.ids_seen = set()
status = Status()
for classifier in status.classifiers.keys():
for rf in status.classifiers[classifier]['reviewed']:
json = Reader.read_reviewed(rf)
self.ids_seen.add(json['origin'])
def run(self, args, opts):
if len(args) < 1:
raise UsageError()
elif len(args) > 1:
raise UsageError(
"running 'scrapy review' with more than one argument is not supported"
)
file_name = args[0]
status = Status()
if len(opts.classifiers) == 0:
opts.classifiers = status.classifiers.keys(
) #If all classifiers are to be used
#Setting up classifiers which are possible
valid_classifiers = defaultdict(
dict) #Dictionary for currently feasible classifiers only
for classifier_name in status.classifiers.keys():
classifications = []
if status.classifiers[classifier_name]['info'][
'settings'] and opts.classifiers.count(
classifier_name) == 1:
valid_classifiers[classifier_name]['classifications'] = \
sorted(status.classifiers[classifier_name]['classifications'])
#Counting files for valid classifiers
no_files = {}
classifiers = valid_classifiers.keys()
for classifier in valid_classifiers.keys():
reviewed = status.classifiers[classifier]['reviewed']
for classification in list(
valid_classifiers[classifier]['classifications']):
no_files[classification] = len([
x for x in reviewed if x.find(os.sep + classification) >= 0
])
items = Reader.read_unreviewed(file_name)
#Confirmation mode
confirmation_mode = False
conf_input = 3
while conf_input > 2:
try:
conf_input = int(
raw_input(
"1. Keep the same\n2. Turn on confirmation mode"))
except:
print "Wrong input"
if conf_input == 2: confirmation_mode = True
#Review of items
n = opts.i_no
while n < len(items):
print "ITEM {0}/{1}".format(n, len(items))
print no_files
item = items[n]
status.item.review(item)
if n >= opts.i_no:
to_write = {}
for classifier in valid_classifiers.keys():
#Loop to ensure a choice
is_a_choice = False
while is_a_choice == False:
prompt = "Pick classification\n"
choices = {}
i = 0
for classification in valid_classifiers[classifier][
'classifications']:
i += 1
choices[i] = classification
prompt += "{0}. {1}\t".format(i, classification)
if i % 3 == 0: prompt += "\n"
try:
choice = int(raw_input(prompt))
except:
print "Wrong input"
if choices.has_key(choice): is_a_choice = True
to_write[classifier] = choices[choice]
confirmed = True
if confirmation_mode:
confirmed = False
print "Choices: {0}".format("\t".join(
to_write[classifier]
for classifier in to_write.keys()))
try:
choice = int(raw_input("1. Confirm \n 2. Reclassify"))
except:
print "Wrong input"
if choice == 1: confirmed = True
if confirmed:
for classifier in to_write.keys():
classifications
classifier_dir = os.path.join(status.data_dir,
classifier)
no_files[to_write[classifier]] += 1
new_f_name = "{0}0{1}.json".format(
to_write[classifier],
no_files[to_write[classifier]])
with open(os.path.join(classifier_dir, new_f_name),
"wb") as new_f:
new_f.write(json.dumps(item))
item['classifications'] = to_write
with open(
os.path.join(status.to_upload_dir,
"{0}.json".format(str(uuid.uuid4()))),
"wb") as upload_f:
upload_f.write(json.dumps(item))
n += 1
if n == len(items): sys.exit()
def run(self, args, opts):
if len(args) < 1:
raise UsageError()
elif len(args) > 1:
raise UsageError("running 'scrapy review' with more than one argument is not supported")
file_name = args[0]
status = Status()
if len(opts.classifiers) == 0: opts.classifiers = status.classifiers.keys() #If all classifiers are to be used
#Setting up classifiers which are possible
valid_classifiers = defaultdict(dict)#Dictionary for currently feasible classifiers only
for classifier_name in status.classifiers.keys():
classifications = []
if status.classifiers[classifier_name]['info']['settings'] and opts.classifiers.count(classifier_name) == 1:
valid_classifiers[classifier_name]['classifications'] = \
sorted(status.classifiers[classifier_name]['classifications'])
#Counting files for valid classifiers
no_files = {}
classifiers = valid_classifiers.keys()
for classifier in valid_classifiers.keys():
reviewed = status.classifiers[classifier]['reviewed']
for classification in list(valid_classifiers[classifier]['classifications']):
no_files[classification] = len([x for x in reviewed if x.find(os.sep + classification) >= 0])
items = Reader.read_unreviewed(file_name)
#Confirmation mode
confirmation_mode = False
conf_input = 3
while conf_input > 2:
try:
conf_input = int(raw_input("1. Keep the same\n2. Turn on confirmation mode"))
except:
print "Wrong input"
if conf_input == 2: confirmation_mode = True
#Review of items
n = opts.i_no
while n < len(items):
print "ITEM {0}/{1}".format(n, len(items))
print no_files
item = items[n]
status.item.review(item)
if n >= opts.i_no:
to_write = {}
for classifier in valid_classifiers.keys():
#Loop to ensure a choice
is_a_choice = False
while is_a_choice == False:
prompt= "Pick classification\n"
choices = {}
i = 0
for classification in valid_classifiers[classifier]['classifications']:
i+=1
choices[i] = classification
prompt+= "{0}. {1}\t".format(i, classification)
if i % 3 == 0: prompt += "\n"
try:
choice = int(raw_input(prompt))
except:
print "Wrong input"
if choices.has_key(choice): is_a_choice = True
to_write[classifier] = choices[choice]
confirmed = True
if confirmation_mode:
confirmed = False
print "Choices: {0}".format("\t".join(to_write[classifier] for classifier in to_write.keys()))
try:
choice = int(raw_input("1. Confirm \n 2. Reclassify"))
except:
print "Wrong input"
if choice == 1: confirmed = True
if confirmed:
for classifier in to_write.keys():
classifications
classifier_dir = os.path.join(status.data_dir, classifier)
no_files[to_write[classifier]]+=1
new_f_name = "{0}0{1}.json".format(to_write[classifier], no_files[to_write[classifier]])
with open(os.path.join(classifier_dir, new_f_name), "wb") as new_f:
new_f.write(json.dumps(item))
item['classifications'] = to_write
with open(os.path.join(status.to_upload_dir, "{0}.json".format(str(uuid.uuid4()))), "wb") as upload_f:
upload_f.write(json.dumps(item))
n+=1
if n == len(items): sys.exit()