prefect-github

import os from prefect import flow from prefect_github import GitHubCredentials from prefect_github.repository import query_repository from prefect_github.mutations import add_star_starrable @flow(log_prints=True) def github_interaction_flow(): # It's recommended to store tokens securely, e.g., in Prefect Blocks # and load them by name. For a quickstart, we use an env var. github_token = os.environ.get('GITHUB_PAT', 'YOUR_GITHUB_TOKEN') if github_token == 'YOUR_GITHUB_TOKEN' or not github_token: print("Please set the GITHUB_PAT environment variable or replace 'YOUR_GITHUB_TOKEN' with a valid GitHub Personal Access Token.") return # Create a GitHubCredentials block (or load an existing one) github_credentials = GitHubCredentials(token=github_token) # Optionally save the block for re-use in the Prefect UI or other flows: # github_credentials.save(name="my-github-pat", overwrite=True) print(f"Querying repository details for PrefectHQ/prefect...") repository_info = query_repository( owner="PrefectHQ", name="prefect", github_credentials=github_credentials, return_fields="id name url", # Request specific fields ) print(f"Repository Name: {repository_info.get('name')}") print(f"Repository URL: {repository_info.get('url')}") repository_id = repository_info.get("id") if repository_id: print(f"Attempting to add a star to {repository_info.get('name')} (ID: {repository_id})...") # Note: Starring requires a PAT with 'public_repo' or 'repo' scope # This action might fail if the token doesn't have sufficient permissions # or if the repository is already starred by the user associated with the token. try: starrable_status = add_star_starrable( starrable_id=repository_id, github_credentials=github_credentials, ) print(f"Star operation successful: {starrable_status.get('starrable', {}).get('viewerHasStarred')}") except Exception as e: print(f"Failed to add star: {e}") else: print("Could not retrieve repository ID.") if __name__ == "__main__": github_interaction_flow()