[RFC PATCH v2 17/17] discord: Add save_email() in database insertion

[Muhammad Rizki <kiizuha@gnuweeb.org>]

Add save_email() to save raw lore emails to the database.

Explanation:
The functionality is like save_discord_mail() but it's still differ.
Save lore emails if it's not exist in database yet, if it does then
ignore.

Signed-off-by: Muhammad Rizki <kiizuha@gnuweeb.org>
---
 .../database/methods/insertion/__init__.py    |  4 +++-
 .../methods/insertion/insert_email.py         | 20 +++++++++++++++++++
 2 files changed, 23 insertions(+), 1 deletion(-)
 create mode 100644 daemon/dscord/database/methods/insertion/insert_email.py

diff --git a/daemon/dscord/database/methods/insertion/__init__.py b/daemon/dscord/database/methods/insertion/__init__.py
index dd3ecf4..5137b84 100644
--- a/daemon/dscord/database/methods/insertion/__init__.py
+++ b/daemon/dscord/database/methods/insertion/__init__.py
@@ -7,10 +7,12 @@
 from .insert_atom import InsertAtom
 from .insert_broadcast import InsertBroadcast
 from .insert_discord import InsertDiscord
+from .insert_email import InsertEmail
 
 
 class Insertion(
 	InsertAtom,
 	InsertBroadcast,
-	InsertDiscord
+	InsertDiscord,
+	InsertEmail
 ): pass
diff --git a/daemon/dscord/database/methods/insertion/insert_email.py b/daemon/dscord/database/methods/insertion/insert_email.py
new file mode 100644
index 0000000..0fab378
--- /dev/null
+++ b/daemon/dscord/database/methods/insertion/insert_email.py
@@ -0,0 +1,20 @@
+from mysql.connector import errors
+from datetime import datetime
+
+
+class InsertEmail:
+
+	def save_email(self, email_msg_id):
+		try:
+			return self.__insert_email(email_msg_id)
+		except errors.IntegrityError:
+			#
+			# Duplicate data, skip!
+			#
+			return None
+
+
+	def __insert_email(self, email_msg_id):
+		q = "INSERT INTO dc_emails (message_id, created_at) VALUES (%s, %s)"
+		self.cur.execute(q, (email_msg_id, datetime.utcnow()))
+		return self.cur.lastrowid
-- 
Muhammad Rizki