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 ""