1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
#
# Authors:
# @author: David Blaisonneau <david.blaisonneau@orange.com>
# @author: Arnaud Morin <arnaud1.morin@orange.com>
#~ from foreman.api import Api
import requests
from bs4 import BeautifulSoup
import sys
import json
class FreeIP:
""" FreeIP return an available IP in the targeted network """
def __init__ (self, login, password):
""" Init: get authenticity token """
with requests.session() as self.session:
try:
#~ 1/ Get login token and authentify
payload = {}
log_soup = BeautifulSoup(self.session.get('https://127.0.0.1/users/login', verify=False).text)
payload['utf8'] = log_soup.findAll('input',attrs={'name':'utf8'})[0].get('value')
payload['authenticity_token'] = log_soup.findAll('input',attrs={'name':'authenticity_token'})[0].get('value')
if payload['authenticity_token'] == None:
raise requests.exceptions.RequestException("Bad catch of authenticity_token")
payload['commit']='Login'
payload['login[login]'] = login
payload['login[password]'] = password
#~ 2/ Log in
r = self.session.post('https://127.0.0.1/users/login', verify=False, data=payload)
if r.status_code != 200:
raise requests.exceptions.RequestException("Bad login or password")
#~ Get token for host creation
log_soup = BeautifulSoup(self.session.get('https://127.0.0.1/hosts/new', verify=False).text)
self.authenticity_token = log_soup.findAll('input',attrs={'name':'authenticity_token'})[0].get('value')
if payload['authenticity_token'] == None:
raise requests.exceptions.RequestException("Bad catch of authenticity_token")
except requests.exceptions.RequestException as e:
print("Error connection Foreman to get a free ip")
print(e)
sys.exit(1)
pass
def get(self, subnet, mac = ""):
payload = {"host_mac": mac, "subnet_id": subnet}
payload['authenticity_token'] = self.authenticity_token
try:
self.last_ip = json.loads(self.session.post('https://127.0.0.1/subnets/freeip', verify=False, data=payload).text)['ip']
if payload['authenticity_token'] == None:
raise requests.exceptions.RequestException("Error getting free IP")
except requests.exceptions.RequestException as e:
print("Error connection Foreman to get a free ip")
print(e)
sys.exit(1)
return self.last_ip
if __name__ == "__main__":
import pprint
import sys
if len(sys.argv) == 4:
f = FreeIP(sys.argv[1], sys.argv[2])
print(f.get(sys.argv[3]))
else:
print('Error: Usage\npython {} foreman_user foreman_password subnet'.format(sys.argv[0]))
|
