Compare commits

...

2 commits

Author SHA1 Message Date
Ruben van de Ven
881fb6104b rm city and country 2021-03-18 14:51:54 +01:00
Ruben van de Ven
25cbebca65 get rid of beginners mistake: credentials in external file 2021-03-17 18:16:06 +01:00
2 changed files with 13 additions and 5 deletions

4
no_credentials.json Normal file
View file

@ -0,0 +1,4 @@
{
"user": "",
"password": ""
}

View file

@ -15,13 +15,10 @@ default_categories = [
'Technology', 'Technology',
'Deployments', 'Deployments',
'Dataset', 'Dataset',
'City', # 'City',
'Country', # 'Country',
] ]
username = "Ruben2@SemanticGraphFetcher"
password = "bdqjse4jodn34rbj73l0agrtb306v693"
parser = argparse.ArgumentParser(description='Turn wiki into nodes & links, usable by d3-force.') parser = argparse.ArgumentParser(description='Turn wiki into nodes & links, usable by d3-force.')
parser.add_argument('--categories', metavar='categories', default=default_categories, nargs='+', parser.add_argument('--categories', metavar='categories', default=default_categories, nargs='+',
help='Categories') help='Categories')
@ -29,9 +26,16 @@ parser.add_argument('--url', default="https://www.securityvision.io/wiki/api.ph
help='Wiki API URL') help='Wiki API URL')
parser.add_argument('--output', default="semantic_data.json", parser.add_argument('--output', default="semantic_data.json",
help='Output JSON file') help='Output JSON file')
parser.add_argument('--credentials', default="no_credentials.json",
help="JSON file containing the Bot's credentials")
args = parser.parse_args() args = parser.parse_args()
with open(args.credentials) as fp:
credentials = json.load(fp)
username = credentials['user']
password = credentials['password']
def get_session(): def get_session():
S = requests.Session() S = requests.Session()