#!/usr/bin/env python
"""
Script to import a list of urls into a BlackList property object. Each
line in the specified text file is added as an item to a property
object.
"""
__version__ = "1.1 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 sys
import packetlogic2
# Handle arguments
try:
[host, user, pwd, filename] = sys.argv[1:]
except ValueError:
print "Usage: %s plhost pluser plpass csvfile" % sys.argv[0]
print "Example: %s 192.168.1.25 admin pldemo00 blacklist.txt" % sys.argv[0]
sys.exit(1)
# Connect to PacketLogic
try:
pl = packetlogic2.connect(host, user, pwd)
r = pl.Ruleset()
except:
t, v, tb = sys.exc_info()
print "Couldn't connect to PacketLogic: %s" % v
sys.exit(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)
# Make sure the object exists
o = r.object_get("/PropertyObjects/BlackList")
if not o:
r.object_add("/PropertyObjects/BlackList")
o = r.object_get("/PropertyObjects/BlackList")
# Clean object from any old items
for item in o.items:
o.remove(item)
# Load items from file
for url in f:
url = url.rstrip()
print "Adding url %s" % url
o.add("URL=%s" % url)
r.commit ()