python读取文件数据堆栈溢出的原因_使用Python格式化堆栈溢出问题

# script.py

# usage: python3 script.py > stack_overflow_question.txt

from os.path import basename

from traceback import format_exc

def stack_overflow_question_generator(q_params):

"takes a source script & descriptions and returns a Stack Overflow question"

# get the code from the source script & pad each line with spaces

with open(q_params['source']) as f:

source = f.readlines()

code = ''.join(4 * ' ' + i for i in source)

# run the source script and get the traceback object

try:

trace_obj = None

try:

source_module = q_params['source'][:q_params['source'].index('.')]

__import__(source_module)

except Exception as e:

raise RuntimeError from e

except RuntimeError:

trace_obj = format_exc()

# raise a warning if the source script doesn't raise an error

if not trace_obj:

no_error_message = 'No exception raised by ' + q_params['source']

raise UserWarning(no_error_message)

# break the trace up into lines

trace_lines = trace_obj.split('\n')

# find the first line in the trace from the source and target it & its index

for i, t in enumerate(trace_lines):

if q_params['source'] in t:

target_line = t

index_one = i + 1

break

# find the first line of the outer trace and target its index

break_trace = ' '.join(['The above exception was the direct',

'cause of the following exception:'])

index_two = trace_lines.index(break_trace) - 1

# find the source script name in the target line & rebuild it with that name

quotes_index = [i for i, x in enumerate(target_line) if x == '"'][:2]

source_name = basename(target_line[quotes_index[0] + 1:quotes_index[1]])

filename = ''.join([' File "', source_name, target_line[quotes_index[1]:]])

# format the source traceback as a string & pad each line with spaces

trace = '\n'.join(4 * ' ' + i

for i in [trace_lines[0],

filename] + trace_lines[index_one:index_two])

# build and return the question formatted as a string

return '\n'.join([q_params['issue'], code, q_params['error'], trace,

q_params['request']])

# Example: set source script & other question params (issue, error, request):

question_params = {

'source': 'script.py',

'issue': ' '.join(['I\'m working on a script that will run another script',

'and automatically generate a formatted Stack Overflow',

'question based on the problem the script is having.',

'I\'d like to be able to provide accurate code samples',

'and tracebacks while avoiding all the copying and',

'pasting. Here\'s what I have so far:\n']),

'error': ' '.join(['However, when the source script doesn\'t actually',

'raise an exception, the question-generator script',

'can\'t really format the question appropriately and',

'just raises a warning:\n']),

'request': ' '.join(['Is there a better way for this script to generate',

'questions when the original script doesn\'t raise',

'any errors but still has some issue? I\'d also love',

'to see any implementations that accomplish the task',

'more efficiently or that improve the quality of the',

'question. (Also, apologies if this question is',

'better suited to stack overflow meta or code review;',

'I figured it made more sense here since I\'m trying',

'to build a tool in python for formatting output from',

'a python script.)'])

}

# Generate & print formatted SO question

print(stack_overflow_question_generator(question_params))