Skip to content

Commit 0cfedb5

Browse files
authored
Merge pull request #9 from MidraLab/feat/skip_zenn_default_tag
feat: Skip zenn default tags when retrieving tags
2 parents c55b00a + 44c06a3 commit 0cfedb5

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

opt/notion_manager.py

+10-1
Original file line numberDiff line numberDiff line change
@@ -9,13 +9,22 @@ def __init__(self, api_key, database_id):
99
def add_article(self, title, url, tags, name, date):
1010
new_page = {
1111
"Title": {"title": [{"text": {"content": title}}]},
12-
"Tags": {"multi_select": [{"name": tag} for tag in tags]},
12+
"Tags": {"multi_select": self.get_tags_and_remove_default_tag(tags)},
1313
"Link": {"url": url},
1414
"Author": {"rich_text": [{"text": {"content": name}}]},
1515
"Date": {"date": {"start": date}}
1616
}
1717
self.notion.pages.create(parent={"database_id": self.database_id}, properties=new_page)
1818

19+
20+
# Get the list of tags with zenn's default tag removed from tags
21+
def get_tags_and_remove_default_tag(self, tags) -> list:
22+
notion_tags = []
23+
for tag in tags:
24+
if tag not in ["tech", "idea"]:
25+
notion_tags.append({"name": tag})
26+
return notion_tags
27+
1928
def delete_all_pages(self):
2029
pages = self.notion.databases.query(database_id=self.database_id)
2130
for page in pages['results']:

0 commit comments

Comments
 (0)