Pickle-Daten (oder beliebige Daten) in ein Python-Skript einbetten

Question

In Perl hält der Interpreter quasi an, wenn er auf eine Zeile mit

__END__

darin. Dies wird oft verwendet, um beliebige Daten am Ende eines Perl-Skripts einzubetten. Auf diese Weise kann das Perl-Skript Daten abrufen und speichern, die es "in sich selbst" speichert, was sehr schöne Möglichkeiten eröffnet.

In meinem Fall habe ich ein eingelegtes Objekt, das ich irgendwo speichern möchte. Ich kann zwar eine file.pickle-Datei verwenden, aber ich suchte nach einem kompakteren Ansatz (um das Skript leichter verteilen zu können).

Gibt es einen Mechanismus, der es erlaubt, beliebige Daten irgendwie in ein Python-Skript einzubetten?

Answer 1

Mit pickle können Sie auch direkt auf Strings arbeiten.

s = pickle.dumps(obj)
pickle.loads(s)

Wenn Sie das mit """ kombinieren (Strings in dreifachen Anführungszeichen) können Sie problemlos beliebige gepickelte Daten in Ihrer Datei speichern.

Answer 2

Wenn die Daten nicht besonders groß sind (viele K), würde ich sie einfach mit .encode('base64') kodieren und in eine Zeichenkette mit drei Anführungszeichen einfügen, mit .decode('base64'), um die binären Daten zurückzubekommen, und einem pickle.loads()-Aufruf um sie herum.

Answer 3

In Python können Sie """ (Strings in dreifachen Anführungszeichen) verwenden, um lange Textabschnitte in Ihr Programm einzubetten.

In Ihrem Fall sollten Sie jedoch keine Zeit damit verschwenden.

Wenn Sie ein Objekt haben, das Sie gepickelt haben, würden Sie viel, viel glücklicher sein, dass Objekt als Python-Quelle zu entladen und einfach die Quelle enthalten.

El repr Funktion, die auf die meisten Objekte angewendet wird, gibt eine Python-Quelltextversion des Objekts aus. Wenn Sie die Funktion __repr__ für alle Ihre benutzerdefinierten Klassen verwenden, können Sie Ihre Struktur trivialerweise als Python-Source ausgeben.

Wenn Ihre eingelegte Struktur hingegen als Python-Code begann, belassen Sie sie einfach als Python-Code.

Answer 4

Ich habe diesen Code erstellt. Man führt etwas wie python comp.py foofile.tar.gz aus, und es wird decomp.py erstellt, in das der Inhalt von foofile.tar.gz eingebettet ist. Ich glaube aber nicht, dass das unter Windows wirklich portabel ist, wegen des Popens.

import base64
import sys
import subprocess

inf = open(sys.argv[1],"r+b").read()
outs = base64.b64encode(inf)

decomppy = '''#!/usr/bin/python
import base64

def decomp(data):
    fname = "%s"
    outf = open(fname,"w+b")
    outf.write(base64.b64decode(data))
    outf.close()

    # You can put the rest of your code here.
    #Like this, to unzip an archive
    #import subprocess
    #subprocess.Popen("tar xzf " + fname, shell=True)
    #subprocess.Popen("rm " + fname, shell=True)
''' %(sys.argv[1])

taildata = '''uudata = """%s"""
decomp(uudata)
''' %(outs)

outpy = open("decomp.py","w+b")
outpy.write(decomppy)
outpy.write(taildata)
outpy.close()
subprocess.Popen("chmod +x decomp.py",shell=True)