CODEWITHSUNDEEP
pythonfile
Nick Humrich
Nick Humrich

Reputation: 15755

Deleting n number of lines after specific line of file in python

I am trying to remove a specific number of lines from a file. These lines always occur after a specific comment line. Anyways, talk is cheap, here is an example of what I have.

FILE: --

randomstuff
randomstuff2
randomstuff3

# my comment
extrastuff
randomstuff2
extrastuff2

#some other comment
randomstuff4

So, I am trying to remove the section after # my comment. Perhaps there is someway to delete a line in r+ mode?

Here is what I have so far

with open(file_name, 'a+') as f:
    for line in f:
        if line == my_comment_text:
            f.seek(len(my_comment_text)*-1, 1) # move cursor back to beginning of line
            counter = 4
        if counter > 0:
            del(line) # is there a way to do this?

Not exactly sure how to do this. How do I remove a specific line? I have looked at this possible dup and can't quite figure out how to do it that way either. The answer recommends you read the file, then you re-write it. The problem with this is they are checking for a specific line when they write. I cant do that exactly, plus I dont like the idea of storing the entire files contents in memory. That would eat up a lot of memory with a large file (since every line has to be stored, rather than one at a time).

Any ideas?

Upvotes: 2

Views: 3406

Answers (3)

user3885927
user3885927

Reputation: 3503

I like the answer form @Ashwini. I was working on the solution also and something like this should work if you are OK to write a new file with filtered lines:

def rewriteByRemovingSomeLines(inputFile, outputFile):
unDesiredLines = []
count = 0
skipping = False
fhIn = open(inputFile, 'r')
line = fhIn.readline()
while(line):
    if line.startswith('#I'):
        unDesiredLines.append(count)
        skipping = True
    while (skipping):
        line = fhIn.readline()
        count = count + 1
        if (line == '\n' or line.startswith('#')):
            skipping=False
        else:
            unDesiredLines.append(count)
    count = count + 1
    line = fhIn.readline()
fhIn.close()

fhIn = open(inputFile, 'r')
count = 0
#Write the desired lines to a new file
fhOut = open(outputFile, 'w')
for line in fhIn:
    if not (count in unDesiredLines):
        fhOut.write(line)
    count = count + 1
fhIn.close()
fhOut.close

Upvotes: 0

Ashwini Chaudhary
Ashwini Chaudhary

Reputation: 250971

You can use the fileinput module for this and open the file in inplace=True mode to allow in-place modification:

import fileinput

counter = 0
for line in fileinput.input('inp.txt', inplace=True):
    if not counter:
        if line.startswith('# my comment'):
            counter = 4
        else:
            print line,
    else:
        counter -= 1

Edit per your comment "Or until a blank line is found":

import fileinput

ignore = False
for line in fileinput.input('inp.txt', inplace=True):
    if not ignore:
        if line.startswith('# my comment'):
            ignore = True
        else:
            print line,
    if ignore and line.isspace():
        ignore = False

Upvotes: 1

JohnB
JohnB

Reputation: 108

You can make a small modification to your code and stream the content from one file to the other very easily.

with open(file_name, 'r') as f:
    with open(second_file_name,'w') a t:
    counter = 0
    for line in f:
        if line == my_comment_text:
            counter = 3             
        elif: counter > 0
            counter -= 1
        else:
            w.write(line)

Upvotes: 0

Related Questions