python3で特定の範囲の文字を抽出したい

Question

前提・実現したいこと

python3とsphinxをつかってドキュメントを作っております。
以下のようなrstテキストにおいて特定の範囲を繰り返し抽出したいのですがうまく行く条件がわかりません。ご教授くださると幸いです。
将来的には抽出後の各配列から"タイトル名の同名.rst"に書き出すプログラムを書きたいです

rstの中身

=============
sphinxタイトル
=============

サブタイトルA      #ここから
=============
#なんとかかんとか
#文字列
.. code-block:: python
    :lineos:
    
    python code

#ここまで抽出したい

サブタイトルB
===========
~~~~~~~
~~~~~

サブタイトルC
===========

期待する出力

サブタイトルA      #ここから
=============
#なんとかかんとか
#文字列
.. code-block:: python
    :lineos:
    
    python code

   #ここまで抽出したい

補足情報（FW/ツールのバージョンなど）

言語:python3

Answer 1

file_names = ['in-rst']

def extracts(data):
    lnums = [i for i,l in enumerate(data) if l.startswith('======')]
    title = data[0:lnums[1]+1]
    results = []
    for i in range(2, len(lnums)-1):
        results.append(''.join(data[lnums[i]-1:lnums[i+1]-2]))
    results.append(''.join(data[lnums[-1]-1:]))
    return title, results

for file_name in file_names:
    with open(file_name, 'r') as f:
        data = f.readlines()
    title, results = extracts(data)
    print(title)
    for item in results:
        print('-'*50 + 'start')
        print(item)
        print('-'*50 + 'end')

in-rstの中身

=============
sphinxタイトル
=============

サブタイトルA      #ここから
=============
#なんとかかんとか
#文字列
.. code-block:: python
    :lineos:

    python code

#ここまで抽出したい

サブタイトルB
===========
~~~~~~~
~~~~~

サブタイトルC
===========

Answer 2

test.txtを対象として、「サブタイトル」を含む行から「#ここまで抽出したい」を含む行までを出力するスクリプトを書いてみました。

追記。変数outfileに書き込み先ファイルを持たせるようにし、併せて出力が必要か否かのフラグの機能も持たせるようにしました。
流れとしては間違っていないと思いますが、細かいミスが残っているかもしれません。

import re
outfile = ""

with open("test.txt", "r") as lines:
  for line in lines:
    line=line.rstrip('\r\n')
    if outfile=="":
      m = re.search("サブタイトル([A-Z]+)",line)
      if m:
        outfile = "out_{}.txt".format(m.group(1))
        fo = open(outfile, "w")
        fo.write(line)
        fo.write("\n")
    else:
      fo.write(line)
      fo.write("\n")
      m = re.match("#ここまで抽出したい",line)
      if m:
        fo.close()
        outfile=""

Comments

[TakamasaIijima]

KojiDoiさん　
回答ありがとうございます。
参考にさせていただきます。
想定していました動作としましては、AからBまでだけではなくBからCまでもといったように一つのセクションづつをすべて取り出したいのですが,正規表現で＝と文字列をうまく使って抽出できたりしますでしょうか..?

Answer 3

頂いた回答を元に元のrstを少しだけ改変して抽出することができました

=============
sphinxタイトル
=============

§サブタイトルA      #ここから->regex replace(Atom側で)で条件を(.+\W)(\n)(\W[==].*)として置換
=============   #置換先文字列を§$1$2$3としてタイトルの頭に§というマーカーを設置
#なんとかかんとか
#文字列
.. code-block:: python
    :lineos:

    python code

#ここまで抽出したい->ここを.*\n.*\n\nで発見させて#---というマーカーを設置

#---


§サブタイトルB
===========
~~~~~~~
~~~~~

#---

§サブタイトルC
===========
...~

以下に抽出用プログラムのコードを記します

import re
import os.path

base = os.path.dirname(os.path.abspath(__file__))
name = os.path.normpath(os.path.join(base, '../対象rstの名前.rst'))

outfile = ""

i = 0

with open(name, "r", encoding="utf-8") as lines:
    for line in lines:
        line = line.rstrip('\n')
        if outfile == "":
            m = re.search("§", line)
            if m:
                i += 1
                outfile = os.path.normpath(os.path.join(base, './result/'+str(i) + line + ".rst"))
                print('sub/3/' +str(i) + line)
                fo = open(outfile, "w", encoding="utf-8")
                fo.write('sub/2/'+line)
                fo.write("\n")
        else:
            fo.write(line)
            fo.write("\n")
            m = re.match("#-.*", line)
            if m:
                fo.close()
                outfile = ""

コードについてアドバイスなどありましたらぜひご教授ください
回答ありがとうございました！
(WindowsのPycharmでopenがUTF-8じゃないのなんとかなんないですかね...Cpとかになっちゃう。。)