mirror of
https://github.com/valitydev/redash.git
synced 2024-11-07 17:38:54 +00:00
73 lines
2.1 KiB
Python
73 lines
2.1 KiB
Python
import json
|
|
|
|
from redash import query_runner
|
|
from redash.models import DataSource
|
|
|
|
|
|
def update(data_source):
|
|
print "[%s] Old options: %s" % (data_source.name, data_source.options)
|
|
|
|
if query_runner.validate_configuration(data_source.type, data_source.options):
|
|
print "[%s] configuration already valid. skipping." % data_source.name
|
|
return
|
|
|
|
if data_source.type == 'pg':
|
|
values = data_source.options.split(" ")
|
|
configuration = {}
|
|
for value in values:
|
|
k, v = value.split("=", 1)
|
|
configuration[k] = v
|
|
if k == 'port':
|
|
configuration[k] = int(v)
|
|
|
|
data_source.options = json.dumps(configuration)
|
|
|
|
elif data_source.type == 'mysql':
|
|
mapping = {
|
|
'Server': 'host',
|
|
'User': 'user',
|
|
'Pwd': 'passwd',
|
|
'Database': 'db'
|
|
}
|
|
|
|
values = data_source.options.split(";")
|
|
configuration = {}
|
|
for value in values:
|
|
k, v = value.split("=", 1)
|
|
configuration[mapping[k]] = v
|
|
data_source.options = json.dumps(configuration)
|
|
|
|
elif data_source.type == 'graphite':
|
|
old_config = json.loads(data_source.options)
|
|
|
|
configuration = {
|
|
"url": old_config["url"]
|
|
}
|
|
|
|
if "verify" in old_config:
|
|
configuration['verify'] = old_config['verify']
|
|
|
|
if "auth" in old_config:
|
|
configuration['username'], configuration['password'] = old_config["auth"]
|
|
|
|
data_source.options = json.dumps(configuration)
|
|
|
|
elif data_source.type == 'url':
|
|
data_source.options = json.dumps({"url": data_source.options})
|
|
|
|
elif data_source.type == 'script':
|
|
data_source.options = json.dumps({"path": data_source.options})
|
|
|
|
elif data_source.type == 'mongo':
|
|
data_source.type = 'mongodb'
|
|
|
|
else:
|
|
print "[%s] No need to convert type of: %s" % (data_source.name, data_source.type)
|
|
|
|
print "[%s] New options: %s" % (data_source.name, data_source.options)
|
|
data_source.save()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
for data_source in DataSource.all():
|
|
update(data_source) |