from mercurial import extensions

def pushnonbookmarked(orig, ui, repo, *args, **opts):

    'Call push with only the heads that are not bookmarked'

    if opts.get('non_bookmarked'):

        if opts.get('rev'):

            heads = [repo.lookup(r) for r in opts.get('rev')]

        else:

            heads = repo.heads()

	markheads = parse(repo).values()

        opts['rev'] = [head for head in heads if not(head in markheads)]

    orig(ui, repo, *args, **opts)

def uisetup(ui):

    'Replace push with a decorator to provide --non-bookmarked option'

    entry = extensions.wrapcommand(commands.table, 'push', pushnonbookmarked)

    entry[1].append(('', 'non-bookmarked', None, _("push all heads that are not bookmarked")))