Use same variable names as testing environment

pull/22/head
Raphaël Vinot 2016-07-01 14:21:48 +02:00
parent 13e0cd0901
commit 836845abde
1 changed files with 2 additions and 4 deletions

View File

@ -2,12 +2,10 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
from pymisp import PyMISP from pymisp import PyMISP
from keys import misp_url, misp_key, misp_verifycert from keys import url, key
import argparse import argparse
import tools import tools
def init(url, key):
return PyMISP(url, key, misp_verifycert, 'json')
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Create a given number of event containing a given number of attributes eachh.') parser = argparse.ArgumentParser(description='Create a given number of event containing a given number of attributes eachh.')
@ -15,7 +13,7 @@ if __name__ == '__main__':
parser.add_argument("-a", "--attribute", type=int, help="Number of attributes per event (default 3000)") parser.add_argument("-a", "--attribute", type=int, help="Number of attributes per event (default 3000)")
args = parser.parse_args() args = parser.parse_args()
misp = init(misp_url, misp_key) misp = PyMISP(url, key, True, 'json')
if args.limit is None: if args.limit is None:
args.limit = 1 args.limit = 1