bkeyonly.py

This is an example of a key only file written in Python.

import sys
import struct
import math
import os
import platform
 
if (platform.system() == "Windows"):
    sys.path.append("C:\\Program Files\\Actian\\Zen\\bin")
else:
    sys.path.append("/usr/local/actianzen/lib64")
 
import btrievePython
 
min_x = 2
max_x = 255
btrieveFileName = "primes.btr"
recordFormat = "<i"
recordLength = 4
 
# If the incorrect number of arguments were given.
if (len(sys.argv) != 2):
    sys.exit("Usage: " + os.path.basename(sys.argv[0]) + " uint8_value_greater_than_one")
 
integerValue = int(sys.argv[1])
 
# If integerValue is out of range.
if (integerValue < min_x) or (integerValue > max_x):
    sys.exit("Usage: " + os.path.basename(sys.argv[0]) + " uint8_value_greater_than_one")
 
btrieveClient = btrievePython.BtrieveClient()
assert(btrieveClient != None)
 
btrieveKeySegment = btrievePython.BtrieveKeySegment()
assert(btrieveKeySegment != None)
 
rc = btrieveKeySegment.SetField(0, recordLength, btrievePython.Btrieve.DATA_TYPE_UNSIGNED_BINARY)
assert(rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR), "Error: BtrieveKeySegment::SetField():%d:%s.\n" % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
 
btrieveIndexAttributes = btrievePython.BtrieveIndexAttributes()
assert(btrieveIndexAttributes != None)
 
rc = btrieveIndexAttributes.AddKeySegment(btrieveKeySegment)
assert(rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR), "Error: BtrieveIndexAttributes::AddKeySegment():%d:%s.\n" % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
 
btrieveFileAttributes = btrievePython.BtrieveFileAttributes()
assert(btrieveFileAttributes != None)
 
 
rc = btrieveFileAttributes.SetKeyOnly(True)
assert(rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR), "Error: BtrieveFileAttributes::SetKeyOnly():%d:%s.\n" % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
 
 
rc = btrieveFileAttributes.SetFixedRecordLength(recordLength)
assert(rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR), "Error: BtrieveFileAttributes::SetFixedRecordLength():%d:%s.\n" % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
 
rc = btrieveClient.FileCreate(btrieveFileAttributes, btrieveIndexAttributes, btrieveFileName, btrievePython.Btrieve.CREATE_MODE_OVERWRITE)
assert(rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR), "Error: BtrieveClient::FileCreate():%d:%s.\n" % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
 
btrieveFile = btrievePython.BtrieveFile()
assert(btrieveFile != None)
 
rc = btrieveClient.FileOpen(btrieveFile, btrieveFileName, None, btrievePython.Btrieve.OPEN_MODE_NORMAL)
assert(rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR), "Error: BtrieveClient::FileOpen():%d:%s.\n" % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
 
# For all the values of x.
for i in range(min_x, max_x + 1):
    record = struct.pack(recordFormat, i)
 
    rc = btrieveFile.RecordCreate(record)
    assert(rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR), "Error: BtrieveFile::RecordCreate():%d:%s.\n"  % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
 
# For all the prime numbers in the range min_x through max_x.
for j in range(2, int(max_x / 2) + 1):
    for i in range(j * 2, max_x + 1, j):
        record = bytes(recordLength)
        key = struct.pack(recordFormat, i)
 
        rc = btrieveFile.RecordRetrieve(btrievePython.Btrieve.COMPARISON_EQUAL, btrievePython.Btrieve.INDEX_1, key, record)
 
        if (rc != recordLength):
            rc = btrieveFile.GetLastStatusCode()
            assert(rc == btrievePython.Btrieve.STATUS_CODE_KEY_VALUE_NOT_FOUND), "Error: BtrieveFile::RecordRetrieve():%d:%s.\n" % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
            continue
 
        
        rc = btrieveFile.RecordDelete()
        assert(rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR), "Error: BtrieveFile::RecordDelete():%d:%s.\n"  % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
        
 
# print record
key = struct.pack(recordFormat, integerValue)
 
 
rc = btrieveFile.KeyRetrieve(btrievePython.Btrieve.COMPARISON_EQUAL, btrievePython.Btrieve.INDEX_1, key)
 
if (rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR):
    print(integerValue, "is prime.")
elif (rc == btrievePython.Btrieve.STATUS_CODE_KEY_VALUE_NOT_FOUND):
    print(integerValue, "is not prime.")
else:
    assert(False), "Error: BtrieveFile::KeyRetrieve():%d:%s.\n" % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
 
 
rc = btrieveClient.FileClose(btrieveFile)
assert(rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR), "Error: BtrieveClient::FileClose():%d:%s.\n" % (rc, btrievePython.Btrieve_StatusCodeToString(rc))
 
rc = btrieveClient.FileDelete(btrieveFileName)
assert(rc == btrievePython.Btrieve.STATUS_CODE_NO_ERROR), "Error: BtrieveClient::FileDelete():%d:%s.\n" % (rc, btrievePython.Btrieve_StatusCodeToString(rc))