Can a python script output its own contents to another file?

Question

I'm exporting some data in my python script to an Excel file and I am wondering if I can also export the contents of the currently-running script to that Excel file. This would save me time so that I do not have to manually copy my script to that Excel file each time I run it.

I have no problems writing to Excel, my only problem is that I do not know if python can write out its own script contents.

import pandas
excel_file = pandas.ExcelFile("my_file.xlsx")
df = excel_file.parse("sheet_name")
data = df.sample(n=5, random_state=1)

# I would prefer to get the script contents here
# script_contents = ?

with pandas.ExcelWriter("output.xlsx") as writer:
    data.to_excel(writer, sheet_name='Data')
    script_contents.to_excel(writer, sheet_name='Script')

EDIT: The possible duplicate isn't a total solution since it only returns a string and my problem needs a DataFrame of the source code. See my solution in the answers below.

It's not clear to me what you mean. Are you trying to write the source code to an Excel file? — juanpa.arrivillaga
– juanpa.arrivillaga, Commented Jun 20, 2019 at 20:11
Yes, all code in the script needs to be exported along with the data. — Christian C
– Christian C, Commented Jun 20, 2019 at 20:15
So, your source code is in some file? Then do you know how to open a file, read the text... etc? — juanpa.arrivillaga
– juanpa.arrivillaga, Commented Jun 20, 2019 at 20:16
Possible duplicate of How to get the current script's code in Python? — Gerges
– Gerges, Commented Jun 20, 2019 at 20:21

Blckknght · Accepted Answer · 2019-06-20 20:23:37Z

1

Most Python modules have a __file__ global variable that specifies the source file they were loaded from. It may be that opening up that file and reading it is the easiest way to get the source of your script, assuming that script was all in a single module.

with open(__file__) as source_file:
    source = source_file.read()

# do whatever you want with the source string

answered Jun 20, 2019 at 20:23

Blckknght

106k11 gold badges135 silver badges188 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

blackbrandt · Accepted Answer · 2019-06-20 20:14:14Z

0

See How can I get the source code of a python function?

If your code is written under a function foo():

import inspect
lines = inspect.getsource(foo)

The variable lines now holds the source code for function foo().

answered Jun 20, 2019 at 20:14

blackbrandt

2,1251 gold badge19 silver badges37 bronze badges

Comments

Christian C · Accepted Answer · 2019-06-20 20:37:15Z

I couldn't find a solution looking at the other answers here but I did find a way for this to work, partially using the link Gerges provided in the comments above.

I used the inspect module to get the script contents and convert it to a DataFrame for export. The solution Here's the working solution:

import pandas
from pandas.compat import StringIO
import inspect

def sample():
    excel_file = pandas.ExcelFile(r"file.xlsx")
    df = excel_file.parse("Sheet Name")
    random_sample = df.sample(n=5, random_state=1)
    return random_sample

my_sample = sample()
script_text = pandas.read_csv(StringIO(inspect.getsource(inspect.getmodule(inspect.currentframe()))), sep="\n")

with pandas.ExcelWriter('output.xlsx') as writer:
    my_sample.to_excel(writer, sheet_name='Sample', index=False)
    script_text.to_excel(writer, sheet_name='Script', index=False)

Collectives™ on Stack Overflow

Can a python script output its own contents to another file?

3 Answers 3

Comments

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related