python：读取文本文件，提取数据

拜托，我对python很陌生， 我正在尝试创建一个在命令行中运行的python文件：

myscript.py hosts.txt tag

for example:
 if tag = Test 
 return wsgwtestmg.mass.gov
 if tag = QA
 return wsgwaltmg.mass.gov

hosts.txt :

    Test :  wsgwtestmg.mass.gov
    QA : wsgwaltmg.mass.gov

myscript.py：

 import sys

    def readArgument():
    print('Number of arguments:', len(sys.argv), 'arguments.')
    identifier = str(sys.argv[1])
    hostTag = str(sys.argv[2])
    extractURLhost(identifier,hostTag)

    def extractURLhost(idt,tag):
    print(idt)
    with open(idt) as f:
        content = f.readlines()
        print(content)

    readArgument()

我的结果是：

Number of arguments: 3 arguments.
hosts.txt

['Test :  wsgwtestmg.mass.gov\n', 'QA : wsgwaltmg.mass.gov\n']

我现在做的对吗？或者有其他简单的方法来做同样的事情

如何使用第三个参数提取url

---

谢谢，非常感谢您的帮助。

一个选项是从文本文件中创建一个词典，并从该词典中获取结果。像这样：

from sys import argv
with open(argv[1]) as f:
    addresses = {i.split(" : ")[0] : i.split(" : ")[1] for i in f}
print(addresses[argv[2]])

---

请问一个问题？请修复您的缩进谢谢你的公告板，它很有效，哇，python非常强大。多谢

import sys

with open('hosts.txt') as f:
    print {x.split(' : ')[0].lower(): x.split(' : ')[1] for x in f.read().splitlines()}.get(sys.argv[1:][0].lower(), 'No host found')