NotionデータベースAPI操作: 主キー(タイトル列)を変更・入れ替えてデータベースを複製する

# Notion API Key
NOTION_TOKEN=your_integration_secret
export NOTION_TOKEN

# ryeの場合
$ rye add notion-client

# pipの場合
$ pip install notion-client

import os
from notion_client import Client

# Secret Integrationキーを利用してNotionクライアントを生成
client = Client(auth=os.environ["NOTION_TOKEN"])

# Get DB info from query, title and id
def get_db_names_and_ids_from_query(client, query: str) -> dict:
    """
    Search for databases in Notion using a query string and return a dictionary containing database names and IDs.

    Args:
        client: The Notion client object.
        query (str): The search query string.

    Returns:
        dict: A dictionary containing 'db_title' and 'db_id' for found databases.
        If no databases are found, returns an empty dictionary.
    """
    results = client.search(query=query).get("results")

    # Dict for db_info. {db_title: db_id}
    db_names_ids = {}
    db_count = 0

    for result in results:
        # Check if the result is a database
        if result["object"] == "database":
            db_count += 1
            # db_title: db_id
            db_names_ids[result["title"][0]["plain_text"]] = result["id"]
    
    if db_count:
        print(f"Found {db_count} databases.")
        return db_names_ids
    else:
        print(f"No databases found with query: {query}")
        return {}

# Get Page info (Up to 100) from a specific DB
def get_pages_info_from_db_id(client, database_id: str) -> list:
    """
    Retrieves all information from a specified Notion database.

    Args:
        client: The Notion client object.
        database_id (str): The ID of the Notion database to query.

    Returns:
        list: A list of dictionaries, each containing all information for a page in the database.
    """
    response = client.databases.query(
        **{
            "database_id": database_id,
        }
    )

    return response["results"]
  

# 実行例: コピー元となるデータベースのページ情報を取得
parent_db_pages = get_all_pages_info_from_db_id(client, PARENT_DB_ID)

import time

# Get All Page info from a specific DB, even though it is more than 100 pages
def get_all_pages_info_from_db_id(self, databaseID: str) -> list:
    """Return the query of all the databases."""
    data = self.databases.query(databaseID)
    # Retrieve the object, has_more, and next_cursor
    database_object = data['object']
    has_more = data['has_more']
    next_cursor = data['next_cursor']
    while has_more == True:
        # Sleep for 2 second to avoid rate limit
        time.sleep(2)
        data_while = self.databases.query(databaseID, start_cursor=next_cursor)
        # Append the results to the data
        for row in data_while['results']:
            data['results'].append(row)
        # Update the has_more and next_cursor
        has_more = data_while['has_more']
        next_cursor = data_while['next_cursor']

    # Create a new database object with the updated data
    new_database = {
        "object": database_object,
        "results": data["results"],
        "next_cursor": next_cursor,
        "has_more": has_more
    }
    
    # Return only the results (pages)
    return new_database["results"]

# データベースIDを取得
people_dbs = get_db_names_and_ids_from_query(client, query="People")

# 取得したデータベースのうち、Peopleデータベースのページ情報をすべて取得
people_db_pages = get_pages_info_from_db_id(
	client, 
	database_id=people_dbs['People']
)

print(people_db_pages['People'])

---- 出力結果 ----
{'object': 'page',
 'id': 'xxxxxxx-xxxxxxxxxx',
     'name': 'python',',
 'created_time': '2024-08-21T09:56:00.000Z',
 'last_edited_time': '2024-08-21T09:56:00.000Z',
 'created_by': {'object': 'user',
  'id': 'xxxxxxx-xxxxxxxxxx2',
     'name': 'python','},
 'last_edited_by': {'object': 'user',
  'id': 'xxxxxxx-xxxxxxxxxx2',
     'name': 'python','},
 'cover': None,
 'icon': None,
 'parent': {'type': 'database_id',
  'database_id': 'xxxxxxx-xxxxxxxxxx2',
     'name': 'python','},
 'archived': False,
 'in_trash': False,
# 各プロパティの種類と内容
 'properties': 
 # マルチセレクト列のプロパティ情報
 {'Tags': {'id': 'xxxx',
   'type': 'multi_select',
   'multi_select': [{'id': 'xxxxxxx-xxxxxxxxxx2',
     'name': 'python',
     'color': 'default'}]},
  # URL列のプロパティ情報
  'Website': {'id': 'xxxxx', 'type': 'url', 'url': None},
  # リッチテキスト列のプロパティ情報
  'username': {'id': 'xxxxxxxx',
   'type': 'rich_text',
   'rich_text': [{'type': 'text',
     'text': {'content': 'alice', 'link': None},
     'annotations': {'bold': False,
      'italic': False,
      'strikethrough': False,
      'underline': False,
      'code': False,
      'color': 'default'},
     'plain_text': 'alice',
     'href': None}]},
  # 主キー列のプロパティ情報
  'Name': {'id': 'title',
   'type': 'title',
   'title': [{'type': 'text',
     'text': {'content': 'Alice Wonder', 'link': None},
     'annotations': {'bold': False,
      'italic': False,
      'strikethrough': False,
      'underline': False,
      'code': False,
      'color': 'default'},
     'plain_text': 'Alice Wonder',
     'href': None}]}},
 'url': 'https://www.notion.so/your-info',
 'public_url': 'https://your-url'}

def create_database(parent_id: str, db_name: str) -> dict:
    """
    parent_id(str): ID of the parent page
    db_name(str): Title of the database
    """
    print(f"\n\nCreate database '{db_name}' in page {parent_id}...")
    properties = {
        "Name": {"title": {}},  # This is a required property
        "Description": {"rich_text": {}},
        "In stock": {"checkbox": {}},
        "Food group": {
            "select": {
                "options": [
                    {"name": "🥦 Vegetable", "color": "green"},
                    {"name": "🍎 Fruit", "color": "red"},
                    {"name": "💪 Protein", "color": "yellow"},
                ]
            }
        },
        "Price": {"number": {"format": "dollar"}},
        "Last ordered": {"date": {}},
        "Store availability": {
            "type": "multi_select",
            "multi_select": {
                "options": [
                    {"name": "Duc Loi Market", "color": "blue"},
                    {"name": "Rainbow Grocery", "color": "gray"},
                    {"name": "Nijiya Market", "color": "purple"},
                    {"name": "Gus's Community Market", "color": "yellow"},
                ]
            },
        },
        "+1": {"people": {}},
        "Photo": {"files": {}},
    }
    title = [{"type": "text", "text": {"content": db_name}}]
    icon = {"type": "emoji", "emoji": "🎉"}
    parent = {"type": "page_id", "page_id": parent_id}
    return notion.databases.create(
        parent=parent, title=title, properties=properties, icon=icon
    )

# コピー元主キー列の値: フルネームを抜き出す
orient_full_name = people_db_pages['page_name']["properties"]["Name"]["title"][0]['text']['content']
# コピー元リッチテキスト列の値: usernameを抜き出す
orient_username = people_db_pages['page_name']["properties"]["username"]["rich_text"][0]['text']['content']

changed_properties = {
        # コピー元では主キーだったNameをリッチテキスト列として入稿
        'Name': {
            'type': 'rich_text',
            'rich_text': [{'type': 'text',
            'text': {'content': orient_full_name, 'link': None},
            }],
        },
        # コピー元ではリッチテキスト列だったusernameを主キー列として入稿
        'username': {'id': 'title',
            'type': 'title',
            'title': [{'type': 'text',
                'text': {'content': orient_username, 'link': None},
            }],
        },
    }

# Add a new row with property info to Notion DB
def add_row_with_property_info_to_notion_database(client, database_id, json_properties):
    """
    Add a new row with property info to Notion DB
    Args:
        *client: The Notion client object.
        *database_id: The ID of the database to add the row to.
        *json_properties: The properties of the new row.
    """
    response = client.pages.create(
        **{
            "parent": { "database_id": database_id },
            # Define the properties of the new page
            "properties": json_properties
        }
    )

    print("add_rows_with_properties_to_notion_database completed")
    print(response) # 追加した内容を表示する

add_row_with_property_info_to_notion_database(
	client, 
	people_dbs['People_2'], 
	changed_properties
)

# データベースIDを取得
people_dbs = get_db_names_and_ids_from_query(client, query="people")

# 取得したデータベースのうち、コピー元:Peopleデータベースのページ情報をすべて取得（100件まで）
people_db_pages = get_pages_info_from_db_id(
	client, 
	database_id=people_dbs['People']
)

# コピー元: PeopleDBに100件以上のレコードがあった場合
people_db_pages = get_all_pages_info_from_db_id(
	client, 
	database_id=people_dbs['People']
)

# List of properties to upload
pages_properties_to_upload = []

# 取得したページの分だけループを回す
for page in people_db_pages:
    orient_full_name = page["properties"]["Name"]["title"][0]['text']['content']
    orient_username = page["properties"]["username"]["rich_text"][0]['text']['content']

    changed_properties = {
        # コピー元では主キーだったNameをリッチテキスト列として入稿
        'Name': {
            'type': 'rich_text',
            'rich_text': [{'type': 'text',
            'text': {'content': orient_full_name, 'link': None},
            }],
        },
        # コピー元ではリッチテキスト列だったusernameを主キー列として入稿
        'username': {'id': 'title',
            'type': 'title',
            'title': [{'type': 'text',
                'text': {'content': orient_username, 'link': None},
            }],
        },
    }
		
		# リストにプロパティを追加 （辞書inリストの作成）
    pages_properties_to_upload.append(changed_properties)

import time

for property in pages_properties_to_upload:
    try:
        add_row_with_property_info_to_notion_database(
	        client, 
	        ENTREPRENEUR_DB_ID, 
	        property
	    )
        time.sleep(2)
    except Exception as e:
        print(e)
        break