import requests
import os

bearer_token = os.environ.get("TWITTER_BEARER_TOKEN")

def search_geo_tweets(query, max_results=10):
    url = "https://api.twitter.com/2/tweets/search/recent"
    
    params = {
        "query": f"{query} has:geo",
        "max_results": max_results,
        "tweet.fields": "created_at,geo",
        "expansions": "author_id,geo.place_id",
        "place.fields": "full_name,id,country,country_code,geo,place_type"
    }
    
    headers = {
        "Authorization": f"Bearer {bearer_token}",
        "User-Agent": "v2RecentSearchPython"
    }

    response = requests.get(url, headers=headers, params=params)
    
    if response.status_code != 200:
        raise Exception(response.status_code, response.text)
    
    return response.json()

# 示例调用
result = search_geo_tweets("football")
print(result)

import requests
import os
import json

bearer_token = os.environ.get("TWITTER_BEARER_TOKEN")

def stream_geo_tweets():
    url = "https://api.twitter.com/2/tweets/search/stream"
    
    params = {
        "tweet.fields": "created_at,geo",
        "expansions": "author_id,geo.place_id",
        "place.fields": "full_name,id,country,country_code,geo,place_type"
    }
    
    headers = {
        "Authorization": f"Bearer {bearer_token}",
        "User-Agent": "v2FilteredStreamPython"
    }

    response = requests.get(url, headers=headers, params=params, stream=True)
    
    for response_line in response.iter_lines():
        if response_line:
            json_response = json.loads(response_line)
            print(json.dumps(json_response, indent=4, sort_keys=True))

# 需要先设置规则
def set_stream_rules():
    url = "https://api.twitter.com/2/tweets/search/stream/rules"
    
    headers = {
        "Authorization": f"Bearer {bearer_token}",
        "Content-type": "application/json"
    }
    
    rule = {
        "add": [
            {"value": "has:geo", "tag": "geo-tagged-tweets"}
        ]
    }
    
    response = requests.post(url, headers=headers, json=rule)
    
    if response.status_code != 201:
        raise Exception(response.status_code, response.text)
    
    return response.json()

# 先设置规则，再开始流式传输
set_stream_rules()
stream_geo_tweets()