python - 将我的代码的所有部分放入优化模型的函数中

Question

我正在编写一个优化模型，我想将我的部分代码放在函数中来做这样的事情，但我不确定如何做到这一点：

for file in files:
    data = extract_data(file)
    model = construct_model(data)
    model.solve()

例如，我将复制/粘贴程序中的前 2 块代码：

第一块：

from lxml import etree
from urllib.request import urlopen
from datetime import*

url = urlopen('https://raw.githubusercontent.com/ArthurK-GH/Instances/main/sprint01.xml')
doc = etree.parse(url)
root = doc.getroot()

第二块：

getStart_date = root[0].text
getEnd_date = root[1].text

start_date = datetime.strptime(getStart_date, '%Y-%m-%d')
end_date = datetime.strptime(getEnd_date, '%Y-%m-%d')
delta = (end_date - start_date)

list_of_dates = []
for i in range(delta.days +1):
    day = start_date + timedelta(days=i)
    list_of_dates.append(day)


delta = (end_date - start_date).days +1
list_of_days = []
for x in range(delta):
    list_of_days.append(x+1)
    
zipbObj = zip(list_of_dates,list_of_days)
DicDatesDays = dict(zipbObj)

我不希望第一个块在函数中，但如果你想运行代码，你将需要它。你能指导我找到一种有效的方法吗？谢谢 ！

标签： pythonfunctionoptimization