将文件输入的标准输出保存到文件

hellb0y77

抱歉,但是我对python很陌生,我需要一个脚本来按模式搜索并将整行替换为一个文件,我已经插入了整个脚本,但是问题出在 with fileinput...

#!/usr/bin/env python3

import json
import requests
import sys
import fileinput

url = 'http://169.254.169.254/latest/meta-data/iam/security-credentials/test'

r = requests.get(url)
accesskey = json.loads(r.content.decode('utf-8'))['AccessKeyId']
secretkey = json.loads(r.content.decode('utf-8'))['SecretAccessKey']

with fileinput.input(files=('./envFile.sh')) as envfile:

  for line in envfile:
    if line.strip().startswith('export AWS_ACCESS_KEY='):
      line = 'AWS_ACCESS_KEY="%s"\n' % (accesskey)
    if line.strip().startswith('export AWS_SECRET_KEY='):
      line = 'AWS_SECRET_KEY="%s"\n' % (secretkey)
    sys.stdout.write(line)

输出为:

AWS_ACCESS_KEY="xxxxxxx"
AWS_SECRET_KEY="xxxxxxxxxxxxxxxxxxxxxxxxxx"

现在,输出正确,但是我必须覆盖文件,该怎么办?

拉克什

使用 inplace=True

例如:

import fileinput

with fileinput.input(files='./envFile.sh', inplace=True) as envfile:
    for line in envfile:
        if line.strip().startswith('export AWS_ACCESS_KEY='):
            print(line.replace(line.strip(), 'AWS_ACCESS_KEY="%s"' % (accesskey))) 
        elif line.strip().startswith('export AWS_SECRET_KEY='):
            print(line.replace(line.strip(), 'AWS_SECRET_KEY="%s"' % (secretkey)))
        else:
            print(line)

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章