sst.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you 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 specif

# a stand alone script to read metadata of a given SSTable
from sstmd import SSTableMetadata
from sstidx import IndexInfo
from sstable import SSTableFileName,SSTableReader
import sstable2json
import argparse
import sys
import os


parser = argparse.ArgumentParser(prog="sst")
parser.add_argument("-m", "--metadata", help="display SSTable metadata", action="store_true")
parser.add_argument("-i", "--index", help="display SSTable index information", action="store_true")
parser.add_argument("-d", "--data", help="display SSTable data in json format", action="store_true")
parser.add_argument("-c", "--cql", help="display SSTable cql rows", action="store_true")
parser.add_argument("-v", "--verbose", help="increase output verbosity", action="store_true")
parser.add_argument("sstable", type=str, help="SSTable file")
args = parser.parse_args()
option = "metadata"
verbose = False
if args.verbose:
    verbose = True
if args.sstable is None:
    print "please specify sstable file"
    sys.exit(1)
filename = args.sstable
sstable = SSTableFileName.parse(filename, verbose)

if args.metadata:
    metadata = SSTableMetadata.parse(sstable.statfile(), sstable.sstversion)
    print metadata
elif args.index:
    index = IndexInfo.parse(sstable.indexfile())
    print index
elif args.data:
    if os.path.isfile(sstable.datafile()) != True:
        print "%s not exists" % sstable.datafile()
        sys.exit(1)

    if os.path.isfile(sstable.datafile()) != True:
        print "%s not exists" % sstable.datafile()
        sys.exit(1)

    compressed = True
    if os.path.isfile(sstable.compfile()) != True:
        compressed = False

    reader = SSTableReader(sstable.indexfile(), sstable.datafile(), sstable.compfile(), compressed, args.cql, args.verbose)
    sstable2json.export(reader)