Hướng dẫn how do you write tail commands in python? - làm thế nào để bạn viết các lệnh đuôi trong python?

Được xuất bản lần đầu tại Ayada.dev

Giới thiệu

Trong bài viết này, chúng tôi sẽ viết một triển khai đơn giản của lệnh tail từ Linux. Chương trình này sẽ lấy một tệp và số nguyên n làm đầu vào và in các dòng ____2 cuối cùng từ tệp. Ngoài ra, mục tiêu của chương trình là không đọc toàn bộ tệp vào bộ nhớ cùng một lúc. Điều này sẽ làm cho bộ nhớ chương trình hiệu quả khi xử lý các tệp rất lớn.

Chi tiết thực hiện

Để thực hiện chương trình này, chúng tôi sẽ sử dụng cấu trúc dữ liệu queue. Chúng tôi sẽ đọc từng dòng tệp và đẩy từng dòng vào hàng đợi. Mỗi lần lặp, chúng tôi sẽ kiểm tra kích thước của hàng đợi. Khi kích thước của hàng đợi lớn hơn hoặc bằng n, chúng tôi sẽ loại bỏ phần tử được chèn lần đầu tiên vào hàng đợi. Vì hàng đợi là cấu trúc dữ liệu FIFO, phần tử lần đầu tiên được chèn vào hàng đợi sẽ là phần tử đầu tiên sẽ được xóa khỏi hàng đợi. Điều này sẽ đảm bảo rằng hàng đợi chỉ chứa các dòng n cuối cùng từ tệp.

Như một phần thưởng, tôi đã sử dụng argparse để xử lý đầu vào từ CLI. Hãy xem xét việc thực hiện chương trình này.

import argparse
import queue


def tail(filename, n):
    q = queue.Queue()
    size = 0

    with open(filename) as fh:
        for line in fh:
            q.put(line.strip())
            if size >= n:
                q.get()
            else:
                size += 1      

    for i in range(size):
        print(q.get())

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Print last n lines from a file.')
    parser.add_argument('file', type=str, help='File to read from')
    parser.add_argument('-n', type=int, default=10, help='The last n lines to be printed')
    args = parser.parse_args()

    tail(args.file, args.n)

Nhập chế độ FullScreenen EXIT Mode FullScreen