import subprocess
import os
import timeit

# 10 MB file
commands = ['/media/vineethpenugonda/bca2c1ba-3126-4bbd-a12e-4cf8c2104a8a', '/home/vineethpenugonda/Desktop/', 'rm -rf output1.img' 'dd if=/dev/zero of=output.dat  bs=1M  count=10']

def read_test_10MB():
    os.chdir(commands[0])
    # bs=1024 count=10240
    subprocess.check_output(['sudo', 'dd', 'if=output10.img', 'of=/home/vineethpenugonda/Desktop/output10_copied.img', 'bs=1024' , 'count=10240', 'iflag=nocache', 'status=progress'])
    os.chdir(commands[1])
    subprocess.check_output(['sudo', 'rm', '-rf', 'output10_copied.img'])

def read_test_50MB():
    os.chdir(commands[0])
    # bs=1024 count=51200
    subprocess.check_output(['sudo', 'dd', 'if=output50.img', 'of=/home/vineethpenugonda/Desktop/output50_copied.img', 'bs=1024' , 'count=51200', 'iflag=nocache', 'status=progress'])
    os.chdir(commands[1])
    subprocess.check_output(['sudo', 'rm', '-rf', 'output50_copied.img'])

def read_test_100MB():
    os.chdir(commands[0])
    subprocess.check_output(['sudo', 'dd', 'if=output100.img', 'of=/home/vineethpenugonda/Desktop/output100_copied.img', 'bs=1024' , 'count=102400', 'iflag=nocache', 'status=progress'])
    os.chdir(commands[1])
    subprocess.check_output(['sudo', 'rm', '-rf', 'output100_copied.img'])

if __name__ == "__main__":
    # 10 MB file
    print("\nTime taken to read 10MB file: " + str(timeit.timeit(read_test_10MB, number=1)) + " seconds\n")
    # 50 MB file
    print("\nTime taken to read 50MB file: " + str(timeit.timeit(read_test_50MB, number=1)) + " seconds\n")
    # 100 MB file
    print("\nTime taken to read 100MB file: " + str(timeit.timeit(read_test_100MB, number=1)) + " seconds\n")