如何在CSV中存储X数量的推文后停止Tweepy？

import tweepy
import csv
import sys

class CustomStreamListener(tweepy.StreamListener):
    def __init__(self, api, csv_writer, max_tweets):
        self.api = api
        self.csv_writer = csv_writer
        self.tweet_count = 0
        self.max_tweets = max_tweets

    def on_status(self, status):
        # 处理推文
        self.csv_writer.writerow([status.id, status.text])
        self.tweet_count += 1

        # 达到指定数量的推文后停止Tweepy
        if self.tweet_count >= self.max_tweets:
            print("已存储足够的推文，停止Tweepy")
            sys.exit()

consumer_key = "YOUR_CONSUMER_KEY"
consumer_secret = "YOUR_CONSUMER_SECRET"
access_token = "YOUR_ACCESS_TOKEN"
access_token_secret = "YOUR_ACCESS_TOKEN_SECRET"

auth = tweepy.OAuthHandler(consumer_key, consumer_secret)
auth.set_access_token(access_token, access_token_secret)

api = tweepy.API(auth)

# 设置要存储的CSV文件路径和文件名
csv_file = "tweets.csv"

# 设置要存储的推文数量
max_tweets = X

with open(csv_file, "w", newline="", encoding="utf-8") as file:
    csv_writer = csv.writer(file)
    csv_writer.writerow(["ID", "Text"])  # 写入CSV文件的标题行

    # 创建自定义的StreamListener实例
    stream_listener = CustomStreamListener(api, csv_writer, max_tweets)

    # 创建Stream对象并启动流式数据获取
    stream = tweepy.Stream(auth=api.auth, listener=stream_listener)
    stream.filter(track=["keyword1", "keyword2"])  # 设置要跟踪的关键词