maloja/malojatime.py

import datetime
from calendar import monthrange
from os.path import commonprefix


FIRST_SCROBBLE = int(datetime.datetime.utcnow().replace(tzinfo=datetime.timezone.utc).timestamp())

def register_scrobbletime(timestamp):
	global FIRST_SCROBBLE
	if timestamp < FIRST_SCROBBLE:
		FIRST_SCROBBLE = int(timestamp)

def start_of_scrobbling():
	global FIRST_SCROBBLE
	f = datetime.datetime.utcfromtimestamp(FIRST_SCROBBLE)
	return [f.year]

def end_of_scrobbling():
	global FIRST_SCROBBLE
	f = datetime.datetime.now()
	return [f.year]



# converts strings and stuff to lists
def time_fix(t):


	if isinstance(t, str):
		tod = datetime.datetime.utcnow()
		months = ["january","february","march","april","may","june","july","august","september","october","november","december"]
		weekdays = ["sunday","monday","tuesday","wednesday","thursday","friday","saturday"]

		if t.lower() in ["today","day"]:
			t = [tod.year,tod.month,tod.day]
		elif t.lower() == "month":
			t = [tod.year,tod.month]
		elif t.lower() == "year":
			t = [tod.year]


		elif t.lower() in months:
			#diff = (tod.month - months.index(t.lower()) - 1)
			month = months.index(t.lower()) + 1
			t = [tod.year,month]
			if month > tod.month: t[0] -= 1
		elif t.lower() in weekdays:
			weekday = weekdays.index(t.lower())
			diff = (tod.isoweekday() - weekday) % 7
			dt = tod - datetime.timedelta(diff)
			t = [dt.year,dt.month,dt.day]

	if isinstance(t,str): t = t.split("/")
	#if isinstance(t,tuple): t = list(t)

	t = [int(p) for p in t]

	return t[:3]

# makes times the same precision level
def time_pad(f,t,full=False):
	f,t = time_fix(f), time_fix(t)
	while (len(f) < len(t)) or (full and len(f) < 3):
		if len(f) == 1: f.append(1)
		elif len(f) == 2: f.append(1)
	while (len(f) > len(t)) or (full and len(t) < 3):
		if len(t) == 1: t.append(12)
		elif len(t) == 2: t.append(monthrange(*t)[1])

	return (f,t)


def time_desc(t,short=False):
	if isinstance(t,int):
		if short:
			now = datetime.datetime.now(tz=datetime.timezone.utc)
			difference = int(now.timestamp() - t)

			if difference < 10: return "just now"
			if difference < 60: return str(difference) + " seconds ago"
			difference = int(difference/60)
			if difference < 60: return str(difference) + " minutes ago" if difference>1 else str(difference) + " minute ago"
			difference = int(difference/60)
			if difference < 24: return str(difference) + " hours ago" if difference>1 else str(difference) + " hour ago"
			difference = int(difference/24)
			timeobject = datetime.datetime.utcfromtimestamp(t)
			if difference < 5: return timeobject.strftime("%A")
			if difference < 31: return str(difference) + " days ago" if difference>1 else str(difference) + " day ago"
			#if difference < 300 and tim.year == now.year: return tim.strftime("%B")
			#if difference < 300: return tim.strftime("%B %Y")

			return timeobject.strftime("%d. %B %Y")
		else:
			timeobject = datetime.datetime.utcfromtimestamp(t)
			return timeobject.strftime("%d. %b %Y %I:%M %p")

	else:
		t = time_fix(t)
		date = [1970,1,1]
		date[:len(t)] = t
		timeobject = datetime.datetime(date[0],date[1],date[2],tzinfo=datetime.timezone.utc)

		nowdate = [1970,1,1]
		nowobject = datetime.datetime.now(tz=datetime.timezone.utc)
		nowdate[:len(t)] = [nowobject.year, nowobject.month, nowobject.day][:len(t)]
		nowobject = datetime.datetime(nowdate[0],nowdate[1],nowdate[2],tzinfo=datetime.timezone.utc)
		if short:
			if len(t) == 3:
				diff = (nowobject - timeobject).days
				if diff == 0: return "Today"
				if diff == 1: return "Yesterday"
				if diff < 7: return timeobject.strftime("%A")
			#elif len(t) == 2:


		if len(t) == 3: return timeobject.strftime("%d. %B %Y")
		if len(t) == 2: return timeobject.strftime("%B %Y")
		if len(t) == 1: return timeobject.strftime("%Y")

def range_desc(since=None,to=None,within=None,short=False):

	# the 'short' var we pass down to some of the time_desc calls is a different one than the one here
	# the function-wide one indicates whether we want the 'in' 'from' etc at the start
	# the other one is if we want exact dates or weekdays etc
	# but we still hand it down because it makes sense


	if within is not None:
		since = within
		to = within
	if since is None:
		sincestr = ""
	if to is None:
		tostr = ""

	if isinstance(since,int) and to is None:
		sincestr = "since " + time_desc(since)
		shortsincestr = sincestr
	elif isinstance(to,int) and since is None:
		tostr = "up until " + time_desc(to)
	elif isinstance(since,int) and not isinstance(to,int):
		sincestr = "from " + time_desc(since)
		shortsincestr = time_desc(since)
		tostr = "to the end of " + time_desc(to)
	elif isinstance(to,int) and not isinstance(since,int):
		sincestr = "from the start of " + time_desc(since)
		shortsincestr = time_desc(since)
		tostr = "to " + time_desc(to)

#	if isinstance(since,int) and isinstance(to,int): result = "from " + time_desc(since) + " to " + time_desc(to)
#	elif isinstance(since,int): result = "from " + time_desc(since) + " to the end of " + time_desc(to)
#	elif isinstance(to,int): result = "from the start of " + time_desc(since) + " to " + time_desc(to)
	else:
		if since is not None and to is not None:
			since,to = time_pad(since,to)
			if since == to:
				if len(since) == 3:
					sincestr = "on " + time_desc(since)
				else:
					sincestr = "in " + time_desc(since)
				shortsincestr = time_desc(since,short=True)
				tostr = ""
			elif _week(since,to):

				sincestr = "in " + _week(since,to)
				shortsincestr = _week(since,to)
				tostr = ""
			else:
				fparts = time_desc(since).split(" ")
				tparts = time_desc(to).split(" ")

				fparts.reverse()
				tparts.reverse()

				fparts = fparts[len(commonprefix([fparts,tparts])):]

				fparts.reverse()
				tparts.reverse()

				sincestr =  "from " + " ".join(fparts)
				shortsincestr = " ".join(fparts)
				tostr = "to " + " ".join(tparts)

		else:
			if since is not None:
				sincestr = "since " + time_desc(since)
				shortsincestr = sincestr
			if to is not None:
				tostr = "up until " + time_desc(to)

	if short: return shortsincestr + " " + tostr
	else: return sincestr + " " + tostr





def time_stamps(since=None,to=None,within=None):


	if within is not None:
		since = within
		to = within




	if (since==None): stamp1 = FIRST_SCROBBLE
	else:
		since = time_fix(since)
		date = [1970,1,1]
		date[:len(since)] = since
		stamp1 = int(datetime.datetime(date[0],date[1],date[2],tzinfo=datetime.timezone.utc).timestamp())

	if (to==None): stamp2 = int(datetime.datetime.utcnow().replace(tzinfo=datetime.timezone.utc).timestamp())
	else:
		to = time_fix(to)
		to = _get_next(to)
		date = [1970,1,1]
		date[:len(to)] = to
		stamp2 = int(datetime.datetime(date[0],date[1],date[2],tzinfo=datetime.timezone.utc).timestamp())


	return (stamp1,stamp2)




def delimit_desc(step="month",stepn=1,trail=1):
	txt = ""
	if stepn is not 1: txt += _num(stepn) + "-"
	txt += {"year":"Yearly","month":"Monthly","day":"Daily"}[step.lower()]
	#if trail is not 1: txt += " " + _num(trail) + "-Trailing"
	if trail is not 1: txt += " Trailing" #we don't need all the info in the title

	return txt


def _week(since,to):
	if len(since) != 3 or len(to) != 3: return False
	dt_since, dt_to = datetime.datetime(*since,tzinfo=datetime.timezone.utc), datetime.datetime(*to,tzinfo=datetime.timezone.utc)
	if (dt_to - dt_since).days != 6: return False
	if dt_since.weekday() != 6: return False

	c = dt_to.isocalendar()[:2]
	return str("Week " + str(c[1]) + " " + str(c[0]))

def _num(i):
	names = ["Zero","One","Two","Three","Four","Five","Six","Seven","Eight","Nine","Ten","Eleven","Twelve"]
	if i < len(names): return names[i]
	else: return str(i)






def ranges(since=None,to=None,within=None,step="month",stepn=1,trail=1,max_=None):

	(firstincluded,lastincluded) = time_stamps(since=since,to=to,within=within)

	d_start = _get_start_of(firstincluded,step)
	d_end = _get_start_of(lastincluded,step)
	d_start = _get_next(d_start,step,stepn)			# first range should end right after the first active scrobbling week / month / whatever relevant step
	d_start = _get_next(d_start,step,stepn * trail * -1)	# go one range back to begin


	i = 0
	d_current = d_start
	while not _is_past(d_current,d_end) and (max_ is None or i < max_):
		d_current_end = _get_end(d_current,step,stepn * trail)
		yield (d_current,d_current_end)
		d_current = _get_next(d_current,step,stepn)
		i += 1



def _get_start_of(timestamp,unit):
	date = datetime.datetime.utcfromtimestamp(timestamp)
	if unit == "year":
		#return [date.year,1,1]
		return [date.year]
	elif unit == "month":
		#return [date.year,date.month,1]
		return [date.year,date.month]
	elif unit == "day":
		return [date.year,date.month,date.day]
	elif unit == "week":
		change = (date.weekday() + 1) % 7
		d = datetime.timedelta(days=change)
		newdate = date - d
		return [newdate.year,newdate.month,newdate.day]

def _get_next(time,unit="auto",step=1):
	result = time[:]
	if unit == "auto":
		# see how long the list is, increment by the last specified unit
		unit = [None,"year","month","day"][len(time)]
	#while len(time) < 3:
	#	time.append(1)

	if unit == "year":
		#return [time[0] + step,time[1],time[2]]
		result[0] += step
		return result
	elif unit == "month":
		#result = [time[0],time[1] + step,time[2]]
		result[1] += step
		while result[1] > 12:
			result[1] -= 12
			result[0] += 1
		while result[1] < 1:
			result[1] += 12
			result[0] -= 1
		return result
	elif unit == "day":
		dt = datetime.datetime(time[0],time[1],time[2])
		d = datetime.timedelta(days=step)
		newdate = dt + d
		return [newdate.year,newdate.month,newdate.day]
		#eugh
	elif unit == "week":
		return _get_next(time,"day",step * 7)

# like _get_next(), but gets the last INCLUDED day / month whatever
def _get_end(time,unit="auto",step=1):
	if step == 1:
		if unit == "auto": return time[:]
		if unit == "year" and len(time) == 1: return time[:]
		if unit == "month" and len(time) == 2: return time[:]
		if unit == "day" and len(time) == 3: return time[:]
	exc = _get_next(time,unit,step)
	inc = _get_next(exc,"auto",-1)
	return inc



def _is_past(date,limit):
	date_, limit_ = date[:], limit[:]
	while len(date_) != 3: date_.append(1)
	while len(limit_) != 3: limit_.append(1)
	if not date_[0] == limit_[0]:
		return date_[0] > limit_[0]
	if not date_[1] == limit_[1]:
		return date_[1] > limit_[1]
	return (date_[2] > limit_[2])