Write a "string" as raw binary into a file Python

Question

I'm trying to write a series of files for testing that I am building from scratch. The output of the data payload builder is of type string, and I'm struggling to get the string written directly to the file.

The payload builder only uses hex values, and simply adds a byte for each iteration.

The 'write' functions I have tried all either fall over the writing of strings, or write the ASCII code for the string, rather than the string its self...

I want to end up with a series of files - with the same filename as the data payload (e.g. file ff.txt contains the byte 0xff

def doMakeData(counter):
    dataPayload = "%X" %counter
    if len(dataPayload)%2==1:
        dataPayload = str('0') + str(dataPayload)
    fileName = path+str(dataPayload)+".txt"
    return dataPayload, fileName

def doFilenameMaker(counter):
    counter += 1
    return counter

def saveFile(dataPayload, fileName):
    # with open(fileName, "w") as text_file:
          # text_file.write("%s"%dataPayload)  #this just writes the ASCII for the string
    f = file(fileName, 'wb')
    dataPayload.write(f) #this also writes the ASCII for the string
    f.close()
    return

if __name__ == "__main__":
    path = "C:\Users\me\Desktop\output\\"
    counter = 0
    iterator = 100
    while counter < iterator:
        counter = doFilenameMaker(counter)
        dataPayload, fileName = doMakeData(counter)
        print type(dataPayload)
        saveFile(dataPayload, fileName)

score 4 · Accepted Answer · answered Aug 19 '12 at 05:31

To write just a byte, use chr(n) to get a byte containing integer n.

Your code can be simplified to:

import os
path = r'C:\Users\me\Desktop\output'
for counter in xrange(100):
    with open(os.path.join(path,'{:02x}.txt'.format(counter)),'wb') as f:
        f.write(chr(counter))

Note use of raw string for the path. If you had a '\r' or '\n' in the string they would be treated as a carriage return or linefeed without using a raw string.

f.write is the method to write to a file. chr(counter) generates the byte. Make sure to write in binary mode 'wb' as well.

Thats so much easier. Thank you. I will pick through your code so I can understand whats going on here. I really appreciate your help. — Jay Gattuso, Aug 19 '12 at 05:40

score 1 · Answer 2 · answered Aug 19 '12 at 05:10

1

dataPayload.write(f) # this fails "AttributeError: 'str' object has no attribute 'write'

Of course it does. You don't write to strings; you write to files:

f.write(dataPayload)

That is to say, write() is a method of file objects, not a method of string objects.

You got this right in the commented-out code just above it; not sure why you switched it around here...

answered Aug 19 '12 at 05:10

kindall

178,883
35
278
309

Ah, thats a foolish mistake, thank you for catching that. However, that still writes the ASCII for the string, not the string as binary (I peeked at the file 0A.txt in a hex editor and can see 0x30 0x41, not 0x0A as I want... ) – Jay Gattuso Aug 19 '12 at 05:14
1

Well, that's because you're formatting them to hexadecimal. Don't do that, just convert the numbers using `chr()` and add them to the string. Or use a `bytearray()` -- it's mutable, so is a little more efficient. – kindall Aug 19 '12 at 05:17
OK. I think I understand what you mean. I will go and have a rummage on those methods. Thanks for your time. I appreciate it. – Jay Gattuso Aug 19 '12 at 05:23

Write a "string" as raw binary into a file Python

2 Answers2

Linked