A TempleOS distro for heretics
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 

134 lines
2.8 KiB

#!/usr/bin/env python3
'''
mfa - minimalist file access for TempleOS
usage:
./mfa.py list <filename> [<local-filename>]
./mfa.py put <filename> [<local-filename>]
./mfa.py command <command>
for scripting:
./mfa.py <script
- use tabs to separate arguments
- Mfa.HC must be running in a loop
example script:
wait 5
test
put B:/AutoOSInstall.HC.Z AutoOSInstall.HC
test
command #include "B:/AutoOSInstall"
test
command OSInstall(FALSE);
test
'''
import socket
import sys
import time
TCP_IP = '127.0.0.1'
TCP_PORT = 7770
CHUNK_SIZE = 8
BAUD = 19200
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect((TCP_IP, TCP_PORT))
def read_line():
s = b''
while True:
ch = sock.recv(1)
if ch == b'\n':
print('<', s.decode())
return s.decode()
else:
s += ch
def read_bytes(count):
s = b''
while len(s) < count:
s += sock.recv(1)
return s
def send(line):
print('>', line)
sock.send((line + "\n").encode())
def test():
send('?')
assert read_line() == '!'
def do_command(*argv):
cmd = argv[0]
if cmd == 'put':
filename = argv[1]
local_filename = argv[2] if len(argv) > 2 else filename
with open(local_filename, 'rb') as f:
data = f.read()
size = len(data)
test()
send('P' + filename)
send('S' + str(size))
while len(data):
sock.send(data[0:CHUNK_SIZE])
data = data[CHUNK_SIZE:]
time.sleep(float(CHUNK_SIZE) * 8 / BAUD)
print('Read', size, 'bytes from', local_filename)
elif cmd == 'list':
filename = argv[1]
local_filename = argv[2] if len(argv) > 2 else filename
send('L' + filename)
next_ = read_line()
assert next_[0] == 'S'
size = int(next_[1:])
size_remaining = size
with open(local_filename, 'wb') as f:
while size_remaining:
read_now = min(size_remaining, 0x1000)
data = read_bytes(read_now)
f.write(data)
size_remaining -= len(data)
print('Written', size, 'bytes to', local_filename)
elif cmd == 'command':
command = argv[1]
test()
send("'" + command)
#elif cmd == 'test':
# test()
elif cmd == 'wait':
secs = float(argv[1])
time.sleep(secs)
else:
raise Exception('Command error: ' + argv[0])
if len(sys.argv) > 1:
do_command(*sys.argv[1:])
else:
try:
while True:
if sys.version_info[0] < 3:
entry = raw_input()
else:
entry = input()
if len(entry) and entry[0] != '#':
items = entry.split('\t')
do_command(*items)
except(EOFError):
pass