def __init__(self):
if len(argv) >= 2:
self.config = importlib.import_module(str(argv[1]))
else:
self.config = importlib.import_module('config')
self.unknown_image_path = os.getcwd() + '/unknown_img'
self.url_image_path = os.getcwd() + '/url_img'
self.success_img_path = os.getcwd() + '/success_img/'
self.need_check_img_path = os.getcwd() + '/need_check_img/'
self.not_find_img_pth = os.getcwd() + '/not_find_img/'
self.raidnearby = rs.RaidNearby()
def restart_nearby(self, id):
time.sleep(1)
try:
raid_nearby = raidnearby.RaidNearby()
except KeyboardInterrupt:
sys.exit(1)
except Exception as e:
LOG.error('Failed to init RaidNearby: {}'.format(e))
self.restart_nearby(id)
return
rn_process = Process(target=raid_nearby.main, args=(
self,
id,
))
rn_process.start()
scale_fort = width / 320
crop = fort_img[int(74 * scale_fort):int(246 * scale_fort),
int(74 * scale_fort):int(144 * scale_fort)]
cv2.imwrite(save_url_full_path, crop)
for fullpath_filename in p.glob('*.png'):
filename = os.path.basename(fullpath_filename)
print('Read', filename)
image_name = filename.split('_')
if image_name[0] == 'Fort':
fort_id, ext = os.path.splitext(image_name[1])
if fort_id.isdecimal() == True:
print('fort_id:', fort_id)
img = cv2.imread(str(fullpath_filename), 3)
nearby = rs.RaidNearby(-1)
gym_image_id = nearby.get_gym_image_id(img)
gym_image_fort_id = db.get_gym_image_fort_id(session, gym_image_id)
if gym_image_fort_id is None:
nearby.detectGym(img)
gym_image_id = nearby.get_gym_image_id(img)
gym_image_fort_id = db.get_gym_image_fort_id(
session, gym_image_id)
if int(fort_id) == int(gym_image_fort_id):
print('This gym image is already trained')
os.remove(fullpath_filename)
else:
unknown_fort_id = db.get_unknown_fort_id(session)
print('gym_images id:', gym_image_id, 'fort_id:',
gym_image_fort_id, 'unknow_fort_id:', unknown_fort_id)
scale_fort = width / 320
crop = fort_img[int(74 * scale_fort):int(246 * scale_fort),
int(74 * scale_fort):int(144 * scale_fort)]
cv2.imwrite(save_url_full_path, crop)
for fullpath_filename in p.glob('*.png'):
filename = os.path.basename(fullpath_filename)
print('Read', filename)
image_name = filename.split('_')
if image_name[0] == 'Fort':
fort_id, ext = os.path.splitext(image_name[1])
if fort_id.isdecimal() == True:
print('fort_id:', fort_id)
img = cv2.imread(str(fullpath_filename), 3)
gym_image_id = rs.RaidNearby(-1).get_gym_image_id(i, mg)
gym_image_fort_id = db.get_gym_image_fort_id(session, gym_image_id)
if int(fort_id) == int(gym_image_fort_id):
print('This gym image is already trained')
os.remove(fullpath_filename)
else:
unknown_fort_id = db.get_unknown_fort_id(session)
print('gym_images id:', gym_image_id, 'fort_id:',
gym_image_fort_id, 'unknow_fort_id:', unknown_fort_id)
if gym_image_fort_id == unknown_fort_id or Force_update == True:
if db.update_gym_image(session, gym_image_id,
fort_id) == True:
fort_result_file = os.getcwd(
) + '/success_img/Fort_' + str(fort_id) + '.png'
shutil.move(fullpath_filename, fort_result_file)
fort_count = fort_count + 1
