You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.8 KiB
60 lines
1.8 KiB
from django.db import models, connection
|
|
|
|
from mydefs import ip2int, MyGenericIPAddressField
|
|
|
|
|
|
class IpPoolItemManager(models.Manager):
|
|
def get_pools(self):
|
|
ips = self.raw(r'SELECT id, ip FROM ip_pool_ippoolitem ORDER BY id')
|
|
ips_len = len(list(ips))
|
|
if ips_len < 1:
|
|
return
|
|
last_dg = ip2int(ips[0].ip)
|
|
start_pool = last_dg
|
|
res = list()
|
|
cnt = 0
|
|
for ip in ips:
|
|
ipnt = ip2int(ip.ip)
|
|
if ipnt > last_dg + 1 or ipnt < last_dg - 1:
|
|
res.append((start_pool, last_dg, cnt))
|
|
start_pool = ipnt
|
|
cnt = 0
|
|
last_dg = ipnt
|
|
cnt += 1
|
|
res.append((start_pool, last_dg, cnt))
|
|
return res
|
|
|
|
def add_pool(self, start_ip, end_ip):
|
|
start_ip = ip2int(start_ip)
|
|
end_ip = ip2int(end_ip)
|
|
|
|
if (end_ip - start_ip) > 5000:
|
|
raise Exception(u'Not add over 5000 ip\'s')
|
|
|
|
sql_strs = map(lambda tip: r"(%d)" % tip, range(start_ip, end_ip + 1))
|
|
sql = r'INSERT INTO ip_pool_ippoolitem (ip) VALUES %s' % r",".join(sql_strs)
|
|
print sql
|
|
|
|
cursor = connection.cursor()
|
|
cursor.execute(sql)
|
|
|
|
def get_free_ip(self):
|
|
sql = r'SELECT ip_pool_ippoolitem.id as id, ip_pool_ippoolitem.ip as ip FROM ip_pool_ippoolitem ' \
|
|
r'LEFT JOIN abonent ON abonent.ip_address_id = ip_pool_ippoolitem.id WHERE ' \
|
|
r'abonent.ip_address_id IS NULL LIMIT 1'
|
|
|
|
rs = self.raw(sql)
|
|
rs_len = len(list(rs))
|
|
return None if rs_len is 0 else rs[0]
|
|
|
|
|
|
class IpPoolItem(models.Model):
|
|
ip = MyGenericIPAddressField()
|
|
|
|
objects = IpPoolItemManager()
|
|
|
|
def int_ip(self):
|
|
return ip2int(self.ip)
|
|
|
|
def __unicode__(self):
|
|
return self.ip
|