Add schedule draft and unschedule draft (#6)

Author: shernshiou

substack/api.py

@@ -1,4 +1,5 @@
 import logging
+from datetime import datetime
 
 import requests
 
@@ -195,6 +196,37 @@ def publish_draft(
 		)
 		return Api._handle_response(response=response)
 
+	def schedule_draft(self, draft, draft_datetime: datetime) -> dict:
+		"""
+
+		Args:
+			draft: draft id
+			draft_datetime: datetime to schedule the draft
+
+		Returns:
+
+		"""
+		response = self._session.post(
+			f"{self.publication_url}/drafts/{draft}/schedule",
+			json={"post_date": draft_datetime.isoformat()}
+		)
+		return Api._handle_response(response=response)
+
+	def unschedule_draft(self, draft) -> dict:
+		"""
+
+		Args:
+			draft: draft id
+
+		Returns:
+
+		"""
+		response = self._session.post(
+			f"{self.publication_url}/drafts/{draft}/schedule",
+			json={"post_date": None}
+		)
+		return Api._handle_response(response=response)
+
 	def get_image(self, image):
 		response = self._session.post(
 			f"{self.publication_url}/image",