#!/usr/bin/env python
# vi: encoding=utf-8 ts=8 sts=4 sw=4 et

import sys, os
sys.path.insert(0, os.environ['HOME']+'/tld-ftp-admin/modules')
import ftptree
from common import checkdir
import ftpio

if len(sys.argv) < 3:
    print "ERR: not enough parameters given"
    print "remove.py tree package1 [package2...]"
    sys.exit(1)

checkdir(sys.argv[1])

ftpio.connect('remove')

if not ftpio.lock(sys.argv[1], True):
    print "ERR: %s tree already locked" % sys.argv[1]
    sys.exit(1)

try:
    tree=ftptree.FtpTree(sys.argv[1])
    #tree.do_checkbuild=False
    pkgs = list(set(sys.argv[2:]))
    tree.mark4removal(pkgs)
    tree.removepkgs()
except ftptree.SomeError:
    # In case of problems we need to unlock the tree before exiting
    ftpio.unlock(sys.argv[1])
    sys.exit(1)

ftpio.unlock(sys.argv[1])