def addFileData(): # jsondata = request.form.get("dna_config"); # 파일 로드 file = request.files['file'] if file.content_type == 'application/vnd.ms-excel': try: datalist = [] spamreader = csv.reader(file) for index, row in enumerate(spamreader): # if index == 0 and isNumberString(row[0]) != True: # continue # if isNumberString(row[0]) == True: # datalist.append(row[0]) # else: # datalist.append(None) if int(row[1]) not in [8, 16, 24, 32]: raise InvalidUsage('Mask value is not valid', status_code=501) try: _pattern = Rules_White_IP() _pattern.ip = row[0] _pattern.mask = row[1] _pattern.description = row[2] db_session.add(_pattern) db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code=501) except Exception as e: raise InvalidUsage('CSV 로딩 실패, ' + e.message, status_code=501) # target link와 데이터 크기 비교 # target_type = request.form.get("target_type") # target_seq = request.form.get("target_seq") # if ValidateTargetSize(datalist.__len__(), target_type, target_seq) == False: # raise InvalidUsage('입력 데이터 사이즈가 다릅니다. ', status_code=501) # try: # _pattern = Rules_White_IP() # _pattern.ip = request.form['pattern'].strip() # _pattern.mask = request.form['mask'].strip() # _pattern.description = request.form['desc'] # db_session.add(_pattern) # db_session.commit() # except Exception as e: # db_session.rollback() # raise InvalidUsage('DB 저장 오류', status_code=501) else: raise InvalidUsage('지원하지 않는 파일 포멧 입니다.', status_code=501) return ""
def addcrawllist(): # try: # _pattern = Rules_Crawl() # _pattern.uri =request.form['pattern'] # _pattern.depth = int(request.form['depth']) # _pattern.desc = request.form['desc'] # _pattern.register_path = request.form['source'] # Rules_Crawl.insertData(_pattern) # # except Exception as e: # raise InvalidUsage('DB 저장 오류', status_code = 501) try: _pattern = Manual_Crawling_Info() # _pattern.type = request.form['type'] _pattern.depth = request.form['depth'].strip() _pattern.url = request.form['url'].strip() # _pattern.mask = request.form['mask'].strip() _pattern.comment = request.form['comment'] db_session.add(_pattern) db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code=501) return ""
def addwhiteip(): exists = Rules_White_IP.query.filter_by( ip=request.form['pattern'].strip()).first() if exists is not None: raise InvalidUsage('중복 IP가 존재합니다.', status_code=501) try: _pattern = Rules_White_IP() _pattern.ip = request.form['pattern'].strip() _pattern.mask = request.form['mask'].strip() _pattern.description = request.form['desc'] db_session.add(_pattern) db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code=501) return ""
def setType_Category_DashboardLinkNP(): # whitelist = wl_maintenance_period.query.filter_by(datatype='days').first() # whitelist.wl_maintenance_period = request.form.get('whitelistvalue') try: db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code=501) return json.dumps({'success':True}), 200, {'ContentType':'application/json'}
def editRestartSchedule(seq): try: schedule = db_session.query(DNA_Schedule).filter_by(id=seq).first() schedule.restart_request = 1 db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code=501) return ""
def editwhiteip(seq): _pattern = db_session.query(Rules_White_IP).filter_by(seq=seq).first() try: _pattern.ip = request.form['pattern'].strip() _pattern.mask = request.form['mask'].strip() _pattern.description = request.form['desc'] db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code=501) return ""
def addipcollection(): exists = Rules_IP_Collection.query.filter_by( ip=request.form['pattern'].strip()).first() if exists is not None: raise InvalidUsage('중복 IP가 존재합니다.', status_code=501) try: _pattern = Rules_IP_Collection() _pattern.ip = request.form['pattern'].strip() _pattern.mask = request.form['mask'].strip() _pattern.detection_point = request.form['detection_point'].strip() _pattern.etc = request.form['etc'].strip() _pattern.description = request.form['description'].strip() _pattern.use_yn = request.form['use_yn'].strip() #_pattern.description = request.form['desc'] db_session.add(_pattern) db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code=501) return ""
def editsnortlist(seq): _pattern = db_session.query(Rules_Snort).filter_by(seq=seq).first() try: _pattern.name = request.form['name'] _pattern.pattern = request.form['pattern'] _pattern.level = int(request.form['level']) _pattern.description = request.form['desc'] _pattern.mod_dt = datetime.datetime.now() _pattern.source = request.form['source'] db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code=501) return ""
def editcrawllist(): try: _pattern = Rules_Crawl() _pattern._id = request.form['seq'] _pattern.uri = request.form['pattern'] _pattern.depth = int(request.form['depth']) _pattern.desc = request.form['desc'] _pattern.register_path = request.form['source'] _pattern._index = request.form['index'] Rules_Crawl.insertData(_pattern) except Exception as e: raise InvalidUsage('DB 저장 오류', status_code=501) return json.dumps({'success': True}), 200, { 'ContentType': 'application/json' } ## Initially "" empty string handled statement. I put 200 OK to look clean in the UI
def editType_Category_url(seq): _pattern = db_session.query(Integrated_Type_Category).filter_by(seq=seq).first() try: _pattern.Field_Name = request.form['Field_Name'].strip() # _pattern.IP_Address = request.form['IP_Address'].strip() _pattern.Description = request.form['Description'].strip() # _pattern.Password = request.form['Password'].strip() # _pattern.description = request.form['desc'] # _pattern.mod_dt = datetime.datetime.now().strftime("%Y-%m-%d %H:%M") db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code = 501) return ""
def editSchedule(seq): try: schedule = db_session.query(DNA_Schedule).filter_by(id=seq).first() schedule.dna_id = request.form['dna_id'] schedule.description = request.form['desc'] schedule.cycle = request.form['cycle_value'] + request.form['cycle_opt'] schedule.start_time = request.form['start_time'] schedule.cre_id = session.get('id') schedule.filter_ip = request.form['filter_ip'] schedule.filter_data_type = request.form['filter_data_type'] db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code=501) return ""
def editipcollection(seq): _pattern = db_session.query(Rules_IP_Collection).filter_by(seq=seq).first() try: _pattern.ip = request.form['pattern'].strip() _pattern.mask = request.form['mask'].strip() _pattern.etc = request.form['etc'].strip() _pattern.detection_point = request.form['detection_point'].strip() _pattern.description = request.form['description'].strip() _pattern.use_yn = request.form['use_yn'].strip() #_pattern.description = request.form['desc'] db_session.commit() except Exception as e: db_session.rollback() raise InvalidUsage('DB 저장 오류', status_code=501) return ""