mirror of
https://github.com/jrnl-org/jrnl.git
synced 2025-05-10 16:48:31 +02:00
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
import sys
|
|
import os
|
|
from tzlocal import get_localzone
|
|
import getpass as gp
|
|
import pytz
|
|
|
|
PY3 = sys.version_info[0] == 3
|
|
PY2 = sys.version_info[0] == 2
|
|
STDIN = sys.stdin
|
|
STDERR = sys.stderr
|
|
STDOUT = sys.stdout
|
|
TEST = False
|
|
__cached_tz = None
|
|
|
|
def getpass(prompt):
|
|
if not TEST:
|
|
return gp.getpass(prompt)
|
|
else:
|
|
return py23_input(prompt)
|
|
|
|
|
|
def u(s):
|
|
"""Mock unicode function for python 2 and 3 compatibility."""
|
|
return s if PY3 or type(s) is unicode else unicode(s, "unicode_escape")
|
|
|
|
def prompt(msg):
|
|
"""Prints a message to the std err stream defined in util."""
|
|
if not msg.endswith("\n"):
|
|
msg += "\n"
|
|
STDERR.write(u(msg))
|
|
|
|
def py23_input(msg):
|
|
STDERR.write(u(msg))
|
|
return STDIN.readline().strip()
|
|
|
|
def get_local_timezone():
|
|
"""Returns the Olson identifier of the local timezone.
|
|
In a happy world, tzlocal.get_localzone would do this, but there's a bug on OS X
|
|
that prevents that right now: https://github.com/regebro/tzlocal/issues/6"""
|
|
global __cached_tz
|
|
if not __cached_tz and "darwin" in sys.platform:
|
|
__cached_tz = os.popen("systemsetup -gettimezone").read().replace("Time Zone: ", "").strip()
|
|
if not __cached_tz or __cached_tz not in pytz.all_timezones_set:
|
|
link = os.readlink("/etc/localtime")
|
|
# This is something like /usr/share/zoneinfo/America/Los_Angeles.
|
|
# Find second / from right and take the substring
|
|
__cached_tz = link[link.rfind('/', 0, link.rfind('/'))+1:]
|
|
if not __cached_tz or __cached_tz not in pytz.all_timezones_set:
|
|
__cached_tz = str(get_localzone())
|
|
if not __cached_tz or __cached_tz not in pytz.all_timezones_set:
|
|
__cached_tz = "UTC"
|
|
return __cached_tz
|