#!/usr/bin/env python
# Licensed under the Apache License, Version 2.0 (the "License"); you may not
# use this file except in compliance with the License. You may obtain a copy
# of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import sqlite3
import sys
from optparse import OptionParser

from swift.cli.info import print_info, InfoSystemExit
from swift.common.exceptions import LockTimeout


def run_print_info(args, opts):
    try:
        print_info('container', *args, **opts)
    except InfoSystemExit:
        sys.exit(1)
    except (sqlite3.OperationalError, LockTimeout) as e:
        if not opts.get('stale_reads_ok'):
            opts['stale_reads_ok'] = True
            print('Warning: Possibly Stale Data')
            run_print_info(args, opts)
            sys.exit(2)
        else:
            print('Container info failed: %s' % e)
            sys.exit(1)


if __name__ == '__main__':
    parser = OptionParser('%prog [options] CONTAINER_DB_FILE')
    parser.add_option(
        '-d', '--swift-dir', default='/etc/swift',
        help="Pass location of swift directory")
    parser.add_option(
        '--drop-prefixes', default=False, action="store_true",
        help="When outputting metadata, drop the per-section common prefixes")
    parser.add_option(
        '-v', '--verbose', default=False, action="store_true",
        help="Show all shard ranges. By default, only the number of shard "
             "ranges is displayed if there are many shards.")

    options, args = parser.parse_args()

    if len(args) != 1:
        sys.exit(parser.print_help())

    run_print_info(args, vars(options))