我正在从 Django shell 运行 Python 脚本。在我添加一个功能之前它工作正常main()
。现在它不起作用。当我执行脚本时,什么都没有发生(没有错误或任何东西)。我究竟做错了什么?
在 Django shell 中,我执行如下脚本:execfile('create_xml.py')
创建_xml.py:
import os
import unicodedata
from django.conf import settings
from django.template.loader import render_to_string
from databank.models import Registration
def create_xml(registrations):
for registration in registrations:
xml = render_to_string('databank/metadata.xml', {'registration': registration})
filename = "metadata%s.xml" % (registration.dataset.dataset_id)
file_path = settings.XML_DATA_FILES
with open(os.path.join(file_path, filename), 'w') as f:
f.write(xml.encode('UTF-8'))
def main():
while True:
print "For which datasets would you like to create XML files?"
dsid = input("1: All datasets\t2: Public datasets only\t3: A specific dataset- ")
if ds == 1:
# all datasets
print "Creating XML files for all datasets."
registrations = Registration.objects.all()
create_xml(registrations)
print "All done!"
elif ds == 2:
# only public datasets
print "Creating XML files for public datasets only."
registrations = Registration.objects.filter(dataset__status='public')
create_xml(registrations)
print "All done!"
elif ds == 3:
dsid = input("Please input a dataset id: ")
try:
r = Registration.objects.get(dataset__dataset_id=dsid)
print "Creating XML file for dataset %d." % (dsid)
registrations = [r]
create_xml(registrations)
print "All done!"
except:
print "Not a valid dataset id. please try again."
else:
print "Not a valid option."
if __name__ == "__main__":
main()