如何在Python语言中检查文件是否为Python文件

Question

我正在开发一个web应用程序，应聘者提交python代码，平台通过由招聘人员编写的单元测试。我使用的是Django，在验证单元测试文件时遇到了问题。如何测试一个文件是否是python文件，以及它的格式是否正确(没有语法或词法错误)？使用Python？

谢谢

Answer 1

您可以使用内置函数compile。将文件内容作为字符串作为第一个参数传递，将从中读取代码的文件(如果不是从文件中读取代码，您可以自己命名)作为第二个参数传递，将mode作为第三个参数传递。

尝试：

with open("full/file/path.py") as f:
    try:
        compile(f.read(), 'your_code_name', 'exec')
    except Exception as ex:
        print(ex)

如果测试文件的内容为：

print 1

compile函数将抛出一个SyntaxError，并打印以下消息：

Missing parentheses in call to 'print'. Did you mean print(1)? (your_code_name, line 1)