def main(): global queue, sep, proc_put_file config = configparser.ConfigParser() config.read(CONFIG_FILE) if (not config) or (not config.sections()): print('No se encontro el archivo de configuracion.') exit(0) ftp_images_folder = config['folders'].get('ftp_images', '') car_images_folder = config['folders'].get('car_images', '.') plates_folder = config['folders'].get('plates', '.') sep = config['folders'].get('folder_sep', '.') car_images_folder = car_images_folder + sep + 'cars_'+dt.now().strftime('%Y-%m-%d') if not os.path.exists(car_images_folder): os.mkdir(car_images_folder) plates_folder = plates_folder + sep + 'plates_' + dt.now().strftime('%Y-%m-%d') if not os.path.exists(plates_folder): os.mkdir(plates_folder) ftp = FTP(url=config['ftp'].get('server_url'), port=int(config['ftp'].get('server_port', '9999')), folder=config['ftp'].get('server_folder', '.'), user=config['ftp'].get('user', 'anonymous'), password=config['ftp'].get('password', '') ) if ftp.connect(): exit(0) if ftp.login(): exit(0) # ftp.change_folder() api = API(api_url=config['api'].get('API_URL', ''), token=config['api'].get('API_TOKEN', ''), ) car_detector = CarDetector(model=os.curdir + sep + config['car_detect'].get('model'), threshold=int(config['car_detect'].get('threshold', '50')), car_percent=float(config['car_detect'].get('car_percent', '4.5')) ) proc_put_file = Process(target=put_file_in_queue, args=(ftp,) ) proc_put_file.start() print('Ready and working..\n') json_result = config['folders'].get('results', '.') + sep + f'plates_result_{dt.now().strftime("%Y-%m-%d")}.json' with open(json_result, 'a') as fp: try: fp.write('[\n') while True: check_folder_day(car_images_folder, 'cars_') check_folder_day(plates_folder, 'plates_') images = get_file_in_queue() if not images: time.sleep(1) continue images_list = [] for image, image_name in images: img = np.frombuffer(image, dtype=np.uint8) img = cv2.imdecode(img, 1) if ftp_images_folder: cv2.imwrite(ftp_images_folder + sep + image_name, img) images_list.append(img) # car_images, dict_unique_cars = car_detector.detect(images_list) dict_unique_cars = car_detector.detect(images_list) for key in dict_unique_cars.keys(): folder = car_images_folder + sep + key + dt.now().strftime('%Y-%m-%d_%H:%M:%S') os.mkdir(folder) for car in dict_unique_cars[key]: car_image_path = folder + sep + 'car_' + str(len(os.listdir(folder))) + '.jpg' if not cv2.imwrite(car_image_path, car): print('No se pudo guardar la imagen en {}'.format(car_image_path)) continue response = api.request(car_image_path) result = response['results'] if 'results' in response else [] if len(result) == 0: print('No results for plate') continue plate, plate_box, right_format = api.improve_plate(result) img_plate = cut_and_save(car_image_path, plate_box, plate, plates_folder) print('Plate detected: {}'.format(plate)) fp.write(str(dict(plate=plate, image_path=car_image_path))) # b64encode(img_plate.tobytes()) fp.flush() if not right_format: continue break except KeyboardInterrupt: fp.write(']\n') raise KeyboardInterrupt
def __init__(self): self.public_api = API()
def upload(request): if request.method == 'POST': # 上传到本地目录 try: path = request.POST.get('path') proxy = request.POST.get('proxy') proxy_host = request.POST.getlist('proxy_host') # 上传到本地 f = request.FILES['file'] df = handle_uploaded_file(f) files = {'file': (f.name, open(df, 'rb'))} params = {'action': 'upload'} # 通过proxy处理文件 proxy_obj = Proxy.objects.get(id=proxy) tnow = datetime.datetime.now() # 调用proxy接口,上传文件 api = API('{0}/v1.0/upload'.format(proxy_obj.url), proxy_obj.username, CRYPTOR.decrypt(proxy_obj.password)) result, code = api.req_post(data=params, files=files) if code != 200: file = File(path=path, proxy=proxy_obj, create_time=tnow, status='01', result="上传文件失败") file.save() raise ServerError(result['messege']) # 上传文件成功之后,调用proxy接口,进行文件上传任务 hosts = [] if not proxy_host: hosts = Asset.objects.all().filter(proxy=proxy_obj) if not hosts: # 没有可执行主机 file = File(path=path, proxy=proxy_obj, create_time=tnow, status='01', result="没有可执行主机") file.save() raise RuntimeError("没有可执行主机") else: for host_id in proxy_host: hosts.append(Asset.objects.get(id=host_id)) host_list = [] resource = [] params = {} trigger_kwargs = {} trigger_kwargs['year'] = tnow.year trigger_kwargs['month'] = tnow.month trigger_kwargs['day'] = tnow.day trigger_kwargs['hour'] = tnow.hour trigger_kwargs['minute'] = tnow.minute+1 trigger_kwargs['second'] = tnow.second params['trigger_kwargs'] = trigger_kwargs params['task_name'] = 'ansible' task_kwargs = {} task_kwargs['module_name'] = 'copy' task_kwargs['module_args'] = 'src={0} dest={1}'.format(result.get('fp'), path) # 构建inventory 和 构建主机list for host in hosts: host_list.append(host.networking.all()[0].ip_address) tmp_d = dict() tmp_d['hostname'] = host.networking.all()[0].ip_address tmp_d['port'] = host.port tmp_d['username'] = host.username tmp_d['password'] = CRYPTOR.decrypt(host.password) # 用于前端确定选择的asset tmp_d['id'] = host.id resource.append(tmp_d) task_kwargs['host_list'] = host_list task_kwargs['resource'] = resource params['task_kwargs'] = task_kwargs # 调用proxy接口,创建任务 api = APIRequest('{0}/v1.0/job'.format(proxy_obj.url), proxy_obj.username, CRYPTOR.decrypt(proxy_obj.password)) result, code = api.req_post(json.dumps(params)) if code != 200: file = File(path=path, proxy=proxy_obj, create_time=tnow, status='01', result="上传文件失败") file.save() else: file = File(path=path, proxy=proxy_obj, task_uuid=result['job']['job_id'], create_time=tnow) file.save() except Exception, e: logger.error(traceback.format_exc()) return HttpResponseRedirect(reverse('file_upload'))