在文件中检查两次重复项-Python

Question

config.yml示例，

DBtables:
  CurrentMinuteLoad:
    CSV_File: trend.csv
    Table_Name: currentminuteload

GUI图像，

这可能不是最干净的途径。

我正在制作一个为我使用的另一个python脚本创建config.yml文件的GUI。

使用pysimplegui，我的按钮无法像我期望的那样运行。当前并准确检查参考名称（例如，此处是CurrentMinuteloAd），如果存在，它将踢回去，但会跳过表格的支票（因此Elif语句被跳过）。添加表仍然有效，我只是没有得到想要的双重检查。另外，我必须在GUI中两次点击“确定”按钮才能工作吗？一个怪异的怪癖，对我来说没有任何意义。

def add_table():
  window2.read()
  with open ("config.yml","r") as h:
    if values['new_ref']  in h.read():
      sg.popup('Reference name already exists')  
    elif values['new_db']  in h.read():
      sg.popup('Table name already exists')
    else:
      with open("config.yml", "a+") as f:
        f.write("\n  " + values['new_ref'] +":")
        f.write("\n    CSV_File:" + values['new_csv'])
        f.write("\n    Table_Name:" + values['new_db'])
        f.close()
        sg.popup('The reference "' + values['new_ref'] + '" has been included and will add the table "' + values['new_db'] + '" to PG Admin during the next scheduled upload')

Answer 1

当您使用h.read（）时，应该保存值，因为它会像流一样读取它，然后随后的调用此方法会导致一个空字符串。

尝试这样的编辑代码：

 with open ("config.yml","r") as h:
    content = h.read()
    if values['new_ref']  in content:
      sg.popup('Reference name already exists')  
    elif values['new_db']  in content:
      sg.popup('Table name already exists')
    else:
        # ...

Answer 2

您应该使用真正的YAML解析器更新YAML文件，这将允许您
要检查重复的值，而不在中使用，这会给您false
当新值是现有值（或键）的子字符串时，阳性。

在下面，我添加了两次值，并显示结果YAML。这
第一次围绕new_ref和new_db找不到
匹配虽然是现有值的子字符串。第二次
使用相同的值，当然在先前添加的
值。

import sys
import ruamel.yaml
from pathlib import Path

def add_table(filename, values, verbose=False):
    error = False
    yaml = ruamel.yaml.YAML()
    data = yaml.load(filename)
    dbtables = data['DBtables']
    if values['new_ref'] in dbtables:
        print(f'Reference name "{values["new_ref"]}" already exists') # use sg.popup in your code
        error = True
    for k, v in dbtables.items():
        if values['new_db'] in v.values():
            print(f'Table name "{values["new_db"]}" already exists')
            error = True
    if error:
        return
    dbtables[values['new_ref']] = d = {}
    for x in ['new_cv', 'new_db']:
        d[x] = values[x]
    yaml.dump(data, filename)
    if verbose:
        sys.stdout.write(filename.read_text())
    

values = dict(new_ref='CurrentMinuteL', new_cv='trend_csv', new_db='currentminutel')
add_table(Path('config.yaml'), values, verbose=True)
print('========')
add_table(Path('config.yaml'), values, verbose=True)

给出：

DBtables:
  CurrentMinuteLoad:
    CSV_File: trend.csv
    Table_Name: currentminuteload
  CurrentMinuteL:
    new_cv: trend_csv
    new_db: currentminutel
========
Reference name "CurrentMinuteL" already exists
Table name "currentminutel" already exists