#!/usr/bin/env python

# Tests:
# Write to current version
# Read from current version
# Read from past version

import persifs, sys, os, time

class Timer:
    def __init__(self):
        self.st = self.et = 0

    def start(self):
        self.st = time.time()

    def stop(self):
        self.et = time.time()

    def secs(self):
        return self.et - self.st

    def perSec(self, n):
        return n / self.secs()

def createFiles(n, key=0):
    files = ["file-%d-%d" % (key,x) for x in range(n)]
    timer = Timer()
    timer.start()
    for f in files:
        file(f, "w").close()
    timer.stop()
    print "Created %d files in %g seconds (%g files/sec)" % \
          (n, timer.secs(), timer.perSec(n))

def readFiles(dir):
    timer = Timer()
    timer.start()
    files = os.listdir(dir)
    timer.stop()
    print "Read %d entries in directory in %g seconds (%g files/sec)" % \
          (len(files), timer.secs(), timer.perSec(len(files)))

def main():
    pd, x, x = persifs.findPersifs()
    os.chdir(os.path.join(pd, persifs.nowDir))
    times = []
    nkeys = 5
    # Create and read from current version test
    for key in range(nkeys):
        createFiles(100,key)
        times.append(time.time())
        readFiles(os.getcwd())
    # Read from past directories
    for ts in times:
        p = persifs.pathAtTime(pd, "", ts)
        readFiles(p)

if __name__ == "__main__":
    main()