From 66f1f915af2e7c4a9eff4e2c8c1c9c3b14fc8e26 Mon Sep 17 00:00:00 2001
From: agp8x <git@agp8x.org>
Date: Thu, 12 Apr 2018 20:55:42 +0200
Subject: [PATCH] Initialer Commit

---
 dudle2csv.py     | 58 ++++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |  3 +++
 2 files changed, 61 insertions(+)
 create mode 100644 dudle2csv.py
 create mode 100644 requirements.txt

diff --git a/dudle2csv.py b/dudle2csv.py
new file mode 100644
index 0000000..b5d6735
--- /dev/null
+++ b/dudle2csv.py
@@ -0,0 +1,58 @@
+import argparse
+from collections import defaultdict
+
+import requests
+from bs4 import BeautifulSoup
+
+OPTIONS = {
+	'✘': -1,
+	'?': 0,
+	'✔': 1,
+}
+
+def title_to_user_date(title):
+	parts = title.split(":")
+	date = ":".join(parts[-2:]).strip()
+	name = ":".join(parts[:-2]).strip()
+	return name, date
+
+def crawl(url):
+	response = requests.get(url)
+	html = BeautifulSoup(response.content.decode('utf8'), 'html5lib')
+	participants = defaultdict(dict)
+	votes = html.select(".vote")
+	dates = []
+	
+	for vote in votes:
+		name, date = title_to_user_date(vote.get('title'))
+		vote = OPTIONS[vote.text]
+		participants[name][date] = vote
+		if date not in dates:
+			dates.append(date)
+	dates = sorted(dates)
+	return participants, dates
+
+def dict_to_csv(participants, dates):
+	csv = []
+	head = ["-", *dates]
+	csv.append(head)
+	for name in sorted(participants):
+		row = [name]
+		for date in dates:
+			row.append(str(participants[name][date]))
+		csv.append(row)
+	return "\n".join([",".join(row) for row in csv])
+
+
+if __name__ == "__main__":
+	parser = argparse.ArgumentParser(description="Convert a dudle poll to csv file")
+	parser.add_argument("url")
+	parser.add_argument("--output", "-o", default=False, help="Output file name")
+	args = parser.parse_args()
+	csv = dict_to_csv(*crawl(args.url))
+	if args.output:
+		with open(args.output, "w") as out:
+			out.write(csv)
+	else:
+		print(csv)
+	# python dudle2csv.py "https://dudle.wiai.de/cgi-bin/Fachschaftsdienst/" -o test.csv
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..8a0ecbc
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,3 @@
+requests==2.18.4
+beautifulsoup4==4.6.0
+html5lib==1.0.1
\ No newline at end of file