#!/usr/bin/env python
"""
Script to remove subscribers from the PacketLogic ruleset. The input
are the same as the "ruleset_csv_import.py" and the csv file should be one that
has been used to add objects and rules.
"""
__version__ = "1.2 2008-11-14 Procera Networks"
###############################################################################
#
# NO WARRANTY
#
# BECAUSE THE PROGRAM IS PROVIDED FREE OF CHARGE, THERE IS NO WARRANTY
# FOR THE PROGRAM, TO THE EXTENT PERMITTED BY APPLICABLE LAW. EXCEPT WHEN
# OTHERWISE STATED IN WRITING THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES
# PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESSED
# OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE ENTIRE RISK AS
# TO THE QUALITY AND PERFORMANCE OF THE PROGRAM IS WITH YOU. SHOULD THE
# PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF ALL NECESSARY SERVICING,
# REPAIR OR CORRECTION.
#
# IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
# WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MAY MODIFY AND/OR
# REDISTRIBUTE THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES,
# INCLUDING ANY GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING
# OUT OF THE USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED
# TO LOSS OF DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY
# YOU OR THIRD PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER
# PROGRAMS), EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGES.
#
###############################################################################
import packetlogic2
import sys
# Handle arguments
try:
[plhost, pluser, plpass, prefix, filename] = sys.argv[1:]
except ValueError:
print "Usage: %s plhost pluser plpass prefix csvfile" % sys.argv[0]
print "Example: %s 192.168.1.25 admin pldemo00 /Users/ users.txt" % sys.argv[0]
sys.exit(1)
# Connect to PacketLogic
try:
pl = packetlogic2.connect(plhost, pluser, plpass)
r = pl.Ruleset()
except:
t, v, tb = sys.exc_info()
print "Couldn't connect to PacketLogic: %s" % v
sys.exit(1)
# Make sure we get sane path
if prefix[0] != '/':
prefix = '/' + prefix
if prefix[-1] != '/':
prefix = prefix + '/'
# Make sure the parent object exists
if prefix != '/':
no = r.object_get("/NetObjects%s" % prefix[:-1])
if not no:
no = r.object_add("/NetObjects%s" % prefix[:-1])
# Open file
if filename == "-":
f = sys.stdin
else:
try:
f = file(filename)
except:
t, v, tb = sys.exc_info()
print "Couldn't open file: %s" % v
sys.exit(1)
for lineno, line in enumerate(f):
lineno += 1
line = line.rstrip()
try:
[cls, ip, name] = line.split(",", 2)
name = name.decode("utf-8")
except:
print "Couldn't parse line %d, ignoring" % lineno
continue
# Delete the shaping rule
sr = r.shapingrule_find(cls)
if sr:
r.shapingrule_remove(sr)
# Delete the shaping object
so = r.shapingobject_find(cls)
if so:
r.shapingobject_remove(so)
# Delete the NetObject
no = r.object_get("/NetObjects%s%s" % (prefix, name.encode("utf-8")))
if no:
try:
r.object_remove(no)
except:
continue
r.commit()