搜索文本文件的特定行,但有一个烦人的扭曲

时间:2017-11-08 14:02:50

标签: python search python-3.6

因此,我的python文件需要能够删除文本文件的行。容易吗?关于这个主题,本网站上有大量已回答的问题,不幸的是,如果两行在字符串E.G中有相同的字符系列,则不包含这些问题。 datafile.txt:

LMNO:ABCD:UVWX
ABCD:EFGH:IJKL
BABC:DEFG:HIJK

如果我想删除第二行,我将不得不搜索它,我的系统工作方式意味着你只能知道冒号之前的第一部分。 E.G:

LMNO:
ABCD:
BABC:

所以,如果我想删除包含ABCD甚至ABCD:的行,它将返回第1行。

这是我目前的代码:

import fileinput, sys
def searchFiles(searchPrompt):
    with open("testfile.txt") as f:
        for num, line in enumerate(f, 1):
            if searchPrompt in line:
                return(num)

def deleteLine(lineNumber):
    for lineNumber, line in enumerate(fileinput.input('testfile.txt', 
    inplace=1)):
        if lineNumber == 0:
            continue
        else:
            sys.stdout.write(line)   

searchPrompt = input("Enter username to delete: ")
searchPrompt = (searchPrompt+":")
print(searchPrompt)
lineNumber = searchFiles(searchPrompt)
deleteLine(lineNumber)

2 个答案:

答案 0 :(得分:0)

我认为这就是你想要的:

import fileinput, sys

def removeFromFile(file, user):
    pattern = user+":"
    f = open(file, "r+")
    lines = f.readlines()
    f.seek(0)
    for line in lines:
        if not line.startswith(pattern):
            f.write(line)
    f.truncate()
    f.close()

username = input("Enter username to delete: ")
removeFromFile("datafile.txt", username)

答案 1 :(得分:0)

你可以在python中使用分隔符作为':#39;

例如:

def searchFiles(searchPrompt):
    with open("testfile.txt") as f:
        for num, line in enumerate(f, 1):
            if searchPrompt == line.split(":")[0]:
                return(num)

def deleteLine(lineNumber):
    for lineNumber, line in enumerate(fileinput.input('testfile.txt', 
inplace=1)):
        if lineNumber == 0:
            continue
        else:
            sys.stdout.write(line)   

searchPrompt = input("Enter username to delete: ")
searchPrompt = searchPrompt.split(":")[0]
print(searchPrompt)
lineNumber = searchFiles(searchPrompt)
deleteLine(lineNumber)