pyo3/examples/word-count/word_count/__init__.py

# -*- coding: utf-8 -*-
from __future__ import absolute_import

from .word_count import WordCounter, count_line

__all__ = ["WordCounter", "count_line", "search_py"]


def search_py(path, needle):
    total = 0
    with open(path, "r") as f:
        for line in f:
            words = line.split(" ")
            for word in words:
                if word == needle:
                    total += 1
    return total