icaledit.py

#!/usr/bin/env python3

import sys
from pathlib import Path
from PyQt6.QtWidgets import QApplication, QMainWindow, QTextEdit, QFileDialog, QMessageBox, QWidget, QVBoxLayout
from PyQt6.QtGui import QIcon, QAction
from PyQt6.QtCore import QDate

import icedit_ui
# calendars
from icalendar import Calendar, Event, vCalAddress, vText
from datetime import date, datetime
from dateutil import parser
from pathlib import Path
import os
import pytz


class MainWindow(QMainWindow, icedit_ui.Ui_MainWindow):
	ecal = Calendar()
	actEventName = ''
	selectedEvent = Event()
	
	def __init__(self, *args, **kwargs):
		super().__init__(*args, **kwargs)
		
		self.setupUi(self)
		self.actionNeu.triggered.connect(self.NewFile)
		self.actionOpen.triggered.connect(self.OpenFile)
		self.actionSpeichern.triggered.connect(self.SaveFile)
		self.actionSpeichern.setEnabled(False)
		self.actionSpeichern_unter.triggered.connect(self.SaveAsFile)
		self.actionBeenden.triggered.connect(self.quit)
		self.okButton.clicked.connect(self.EditOk)
		self.edit_menu = self.menubar.addMenu('&Edit Event(s)')
		self.addevent_menu = self.menubar.addMenu('&Add Event(s)')
		self.addevent_menu.setEnabled(False)
		# edit Events menu
		event_action = QAction("Neuer Termin", self)
		event_action.triggered.connect(self.AddEvents)
		self.addevent_menu.addAction(event_action)
		
	def message(self, s):
		self.plainTextEdit.appendPlainText(s)
	
	def OpenFile(self):
		self.filename, ok = QFileDialog.getOpenFileName(self, 'Open File', '', 'ics Files (*.ics)')
		if self.filename:
			self.edit_menu.clear()	#	important !
			fileName = Path(self.filename)	#.selectedFiles()
			print("open ",fileName)
			e = open(fileName, 'rb')
			self.ecal = Calendar.from_ical(e.read())
			self.UpdateInfoPane()
			e.close()
			self.addevent_menu.setEnabled(True)
			self.actionSpeichern.setEnabled(True)
			
	def UpdateInfoPane(self):
		self.plainTextEdit.clear()
		self.edit_menu.clear()	#	important !
		if (self.filename):
			self.message('Datei:\t'+self.filename)
		enum = 1
		for component in self.ecal.walk():
			if component.name == "VEVENT":
				event = Event()
				event = component
				#if not str(component.get("name")):
				if (str(event.get("summary")) == "None"):
					eventId = 'Event '+str(enum)
				else:
					eventId = str(event.get("summary"))
				
				# edit Events menu
				event_action = QAction(eventId, self)
				event_action.triggered.connect(self.EditEvent)
				self.edit_menu.addAction(event_action)

				#print ("eventId: ", eventId)
				event["summary"] = eventId	# finally, that's it :-)
				event["name"] = eventId	# finally, that's it :-)
				#self.message('eventId:\t'+eventId)
				#self.message('Event:\t'+str(event.get("name")))
				self.message('Summary:\t'+str(event.get("summary")))
				self.message('Descr.:\t'+str(event.get("description")))
				self.message('Orga:\t'+str(event.get("organizer")))
				self.message('Ort:\t'+str(event.get("location")))
				if len(str(event.get("url"))):
					self.message('URL:\t'+str(event.get("url")))
				#self.message('GEO:\t'+str(event.get("geo")))
				dtstart = event.get("dtstart")	# .dt is icalendar date object !
				try:
					dt = dtstart.dt
				except:
					dt = dtstart
				if (dt):
					self.message('Start:\t'+dt.strftime("%d.%m.%Y"))	#	+str(type(dtstart)))	#	event.get("dtstart").dt))
				dtend = event.get("dtend")
				try:
					dt = dtend.dt
				except:
					dt = dtend
				if (dt):
					self.message('End:\t'+dt.strftime("%d.%m.%Y"))
				self.message('\t')
				enum += 1
					
	def NewFile(self):
		self.filename, ok = QFileDialog.getSaveFileName(self,  'New File', '', 'ics Files (*.ics)')
		fileName = Path(self.filename)
		if (ok):
			self.ecal = Calendar()	# new empty Calendar
			self.ecal.add("prodid", "-//hoernerfranzracing//DE")	# TODO: get prodid from yaml cfg File
			self.ecal.add("version", "2.0")

			event = Event()
			event.add("summary","New")
			event.add("dtstart",date.today())
			event.add("dtend",date.today())
			event.add("location","")
			event.add("name","New")
			#	event.add("geo","37.386013;-122.082932")
			#	event.add("url","")
			today = QDate.currentDate()
			self.ecal.add_component(event)
			with open(fileName, "wb") as f:
			    f.write(self.ecal.to_ical())
			#print(self.ecal)
			print("new file ",fileName)
			self.UpdateInfoPane()
			self.addevent_menu.setEnabled(True)
			self.actionSpeichern.setEnabled(True)
	
	def AddEvents(self):
		selectedEvent = self.sender()
		self.edit_menu.clear()	#	important !
		print("Sender: ", selectedEvent.text())
		if (self.ecal):
			event = Event()
			for component in self.ecal.walk():
				if component.name == "VEVENT":
					event = component
					start = event.get("dtstart")	# get a valid dtstart
					#print("dtstart: ", start.decoded())
			event = Event()	# new Event !
			event["summary"] = selectedEvent.text()
			event["name"] = event["summary"]
			try:
				event["dtstart"] = start	#	these 2 must be populated !
			except:
				dtstart = date.today()	#	workaround for datetime.datetime issue
			try:
				event["dtend"] = start
			except:
				dtend = date.today()	#	workaround for datetime.datetime issue
			self.ecal.add_component(event)	# kompletten event zum Kalender hinzufügen !
			event = Event()
			self.edit_menu.clear()	#	important !
			self.EditEvent()	# new 02.02.26 - directly edit added Event
			for component in self.ecal.walk():
				if component.name == "VEVENT":
					event = component
					print("Event added: ", event.get("summary"))
					#edit Events menu
					event_action = QAction(event.get("summary"), self)
					event_action.triggered.connect(self.EditEvent)
					self.edit_menu.addAction(event_action)

	def SaveFile(self):
		fileName = Path(self.filename)
		if (self.ecal):
			with open(fileName, "wb") as f:
			    f.write(self.ecal.to_ical())
			
			# workaround for dtstart/dtend type confusion (datetime <-> vDDDTypes):
			e = open(fileName, 'r')	# read as text, NOT binary
			text = e.read()
			e.close()
			text = text.split("\n")
			of = open(fileName, "w")
			for line in text:
				if (line.startswith("DTSTART")) or (line.startswith("DTEND")):
					line = line.replace('-', '')
					line = line.replace(' ', 'T')
					line = line.replace(':', ';')
					line = line.replace(';', ':', 1)
					line = line.replace(';', '')
					line = line.replace('T000000', '')
					text = line.split(":");
					if (line.startswith("DTSTART")):
						dtstart = text[1];
						of.write(line+"\n")
					if (line.startswith("DTEND")):
						dtend = text[1];
						if (dtstart != dtend): # do not write DTEND if if DSTART == DTEND !
							of.write(line+"\n")
				elif (line.startswith("URL")):
					arr = line.split(":");
					url = arr[1].strip()
					if (url):
						of.write(line+"\n")
				else:
					of.write(line+"\n")
			of.close()
			#print(fileName)
		self.UpdateInfoPane()
	
	def SaveAsFile(self):
		self.filename, ok = QFileDialog.getSaveFileName(self,  'Save as File', '', 'ics Files (*.ics)')
		self.SaveFile()
			
	def EditEvent(self):
		# see https://stackoverflow.com/questions/52526040/how-to-get-the-name-of-a-qmenu-item-when-clicked
		selectedEvent = self.sender()
		print("selected Event (edit): ",selectedEvent.text())

		for component in self.ecal.walk():
			if component.name == "VEVENT":
				event = Event()
				event = component
				eventId = str(event.get("summary"))
				print("eventId: ", eventId)
				if (eventId == selectedEvent.text()):
					self.selectedEvent = event
					#print("Found: ", eventId)
					break
		print("selected eventId: ", eventId)
		#self.nameEdit.setEnabled(True)
		self.sumEdit.setEnabled(True)
		self.nameEdit.setText(eventId)
		self.sumEdit.setText(eventId)
		self.startEdit.setEnabled(True)
		try:
			dtstart = event.get("dtstart").dt
		except:
			dtstart = date.today()	#	workaround for datetime.datetime issue
		self.startEdit.setDate(dtstart)
		self.startEdit.setDisplayFormat('dd.MM.yyyy')
		try:
			dtend = event.get("dtend").dt
		except:
			dtend = date.today()
		self.endEdit.setDate(dtend)	#	evEnd)
		self.endEdit.setDisplayFormat('dd.MM.yyyy')
		self.endEdit.setEnabled(True)
		self.urlEdit.setEnabled(True)
		self.urlEdit.setText(event.get("url"))
		self.descEdit.setEnabled(True)
		self.descEdit.setText(event.get("description"))
		self.sumEdit.setEnabled(True)
		self.sumEdit.setText(event.get("summary"))
		self.urlEdit.setEnabled(True)
		self.urlEdit.setText(event.get("url"))
		self.locEdit.setEnabled(True)
		self.locEdit.setText(event.get("location"))
		#self.geoEdit.setEnabled(True)
		#self.geoEdit.setText(event.get("geo"))
		self.okButton.setEnabled(True)
		self.cancelButton.setEnabled(True)
		
	def EditOk(self):
		event = Event()
		for component in self.ecal.walk():
			if component.name == "VEVENT":
				event = Event()
				event = component
				eventId = str(event.get("summary"))
				if (eventId == self.sumEdit.text()):
					self.selectedEvent = event
					#print("Found: ", eventId)
					break
		print("selected eventId: ", eventId)
		self.selectedEvent["summary"] = self.sumEdit.text()
		self.selectedEvent["name"] = self.sumEdit.text()
		self.selectedEvent["description"] = self.descEdit.text()
		startline = self.startEdit.text()
		#print(startline)
		datetime_obj = datetime.strptime(startline, "%d.%m.%Y")
		#print(type(datetime_obj))
		#date = datetime_obj.date()
		#print(type(date))
		self.selectedEvent["dtstart"] = datetime_obj
		startline = self.endEdit.text()
		datetime_obj = datetime.strptime(startline, "%d.%m.%Y")
		#print(type(date))
		#print(vDDDTypes(datetime_obj))
		#date = datetime_obj.date()
		#print(type(date))
		self.selectedEvent["dtend"] = datetime_obj
		
		self.selectedEvent["location"] = self.locEdit.text()
		if len(self.urlEdit.text().strip()):
			self.selectedEvent["url"] = self.urlEdit.text()
		else:
			self.selectedEvent["url"] = ''
		#if not self.geoEdit.text() == "":
		#	self.selectedEvent.add("geo", self.geoEdit.text())
		self.selectedEvent["summary"] = self.sumEdit.text()
		self.selectedEvent["last-modified"] = datetime.today()
		self.UpdateInfoPane()
				
	def quit(self):
		app.quit()


app = QApplication(sys.argv)

w = MainWindow()
title = "icaledit.py v 0.1.3 (C) Werner Joss 2026"
w.setWindowTitle(title)

w.show()

app.setStyle("Fusion")
app.exec()