from PythonQt.QtCore import QSettings
from PythonQt.QtGui import QDesktopServices
import os, string, base64
try:
	from md5 import md5
except ImportError:
	from hashlib import md5 #md5 has been moved to hashlib in python 3
from random import randint
from collections import defaultdict
from time import strftime, localtime

def getScreenshotFormat():
	settings = QSettings()
	settings.beginGroup("main")
	format = settings.value("format", "png")
	settings.endGroup()
	return format

def formatFilename(nameFormat, includeFileExtension = True, custom_vars = dict()):
	try:
		name = strftime(nameFormat.encode('utf-8'), localtime()).decode('utf-8')
	except TypeError:
		name = strftime(nameFormat, localtime()) #fix for python 3
	except ValueError:
		name = nameFormat
		pass
	random_hash = md5(os.urandom(128)).hexdigest()
	random_num = str(randint(0,9))
	random_short = base64.urlsafe_b64encode(os.urandom(6)).decode('utf-8')
	var_dict = defaultdict(str, rnd = random_num, rnd_h = random_hash, rnd_s = random_short)
	var_dict.update(custom_vars)
	try:
		name = string.Formatter().vformat(name, (), var_dict)
	except ValueError:
		pass
	extension = "." + getScreenshotFormat()
	if(includeFileExtension and extension not in name):
		name += extension
	return name

def getPluginDir():
	try:
		return QDesktopServices.storageLocation(QDesktopServices.DataLocation) + "/plugins"
	except AttributeError:
		from PythonQt.QtCore import QStandardPaths
		return QStandardPaths.writableLocation(QStandardPaths.DataLocation) + "/plugins"

def setUrl(url):
	global clipboardUrl
	clipboardUrl = url

def setError(err):
	global uploadingError
	uploadingError = err
