MutualStore/diskUtils.py~ at main · jmlemaster86/MutualStore · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
blockSize = 256
blockNum = 10
diskSize = blockSize * blockNum

def fdisk():
    with open('disk.bin', 'wb') as disk:
        disk.write(bytearray(diskSize))

def saveBlock(loc, data):
    with open('disk.bin', 'r+b') as disk:
        disk.seek(loc * blockSize)
        disk.write(data)

def loadBlock(loc):
    with open('disk.bin', 'rb') as disk:
        disk.seek(loc * blockSize)
        return disk.read(blockSize)

def writeFile(data):
    loc = findEmptyBlock()
    saveBlock(loc, data)

def readFile(loc):
    result = bytearray()
    block = loadBlock(loc)
    while(block[0] != 0):
        for byte in block:
            result.append(byte)
        loc = loc + 1
        block = loadBlock(loc)
    return result


def findEmptyBlock():
    for a in range(blockNum):
        block = loadBlock(a)
        if block[0] != 0:
            return a

if __name__ == "__main__":
    fdisk()
    inputArray = bytearray()
    while True:
        try:
            currentInput = raw_input()
            for a in currentInput:
                inputArray.append(a)
        except:
            break
    writeFile(inputArray)
    print(readFile(0))