2019-05-02 00:39:39 +00:00
|
|
|
import hashlib
|
2018-05-13 12:58:28 +00:00
|
|
|
import json
|
|
|
|
|
2018-05-13 12:44:22 +00:00
|
|
|
from sanic import response
|
2018-05-13 12:58:28 +00:00
|
|
|
|
2019-05-02 00:39:39 +00:00
|
|
|
from datasette.utils import (
|
|
|
|
CustomJSONEncoder,
|
|
|
|
InterruptedError,
|
|
|
|
detect_primary_keys,
|
|
|
|
detect_fts,
|
|
|
|
)
|
2018-05-13 12:44:22 +00:00
|
|
|
from datasette.version import __version__
|
2018-05-13 12:58:28 +00:00
|
|
|
|
|
|
|
from .base import HASH_LENGTH, RenderMixin
|
2018-05-13 12:44:22 +00:00
|
|
|
|
|
|
|
|
|
|
|
class IndexView(RenderMixin):
|
2019-05-04 02:15:14 +00:00
|
|
|
name = "index"
|
2018-05-13 12:55:15 +00:00
|
|
|
|
2018-05-13 12:44:22 +00:00
|
|
|
def __init__(self, datasette):
|
|
|
|
self.ds = datasette
|
|
|
|
|
2018-06-15 06:51:23 +00:00
|
|
|
async def get(self, request, as_format):
|
2018-05-13 12:44:22 +00:00
|
|
|
databases = []
|
2019-05-02 00:39:39 +00:00
|
|
|
for name, db in self.ds.databases.items():
|
|
|
|
table_counts = await db.table_counts(5)
|
|
|
|
views = await db.view_names()
|
|
|
|
tables = {}
|
|
|
|
hidden_table_names = set(await db.hidden_table_names())
|
|
|
|
for table in table_counts:
|
|
|
|
table_columns = await self.ds.table_columns(name, table)
|
|
|
|
tables[table] = {
|
|
|
|
"name": table,
|
|
|
|
"columns": table_columns,
|
|
|
|
"primary_keys": await self.ds.execute_against_connection_in_thread(
|
|
|
|
name, lambda conn: detect_primary_keys(conn, table)
|
|
|
|
),
|
|
|
|
"count": table_counts[table],
|
|
|
|
"hidden": table in hidden_table_names,
|
|
|
|
"fts_table": await self.ds.execute_against_connection_in_thread(
|
|
|
|
name, lambda conn: detect_fts(conn, table)
|
|
|
|
),
|
|
|
|
}
|
|
|
|
hidden_tables = [t for t in tables.values() if t["hidden"]]
|
|
|
|
|
2019-05-04 02:15:14 +00:00
|
|
|
databases.append(
|
|
|
|
{
|
|
|
|
"name": name,
|
|
|
|
"hash": db.hash,
|
|
|
|
"color": db.hash[:6]
|
|
|
|
if db.hash
|
|
|
|
else hashlib.md5(name.encode("utf8")).hexdigest()[:6],
|
|
|
|
"path": self.database_url(name),
|
|
|
|
"tables_truncated": sorted(
|
|
|
|
tables.values(), key=lambda t: t["count"] or 0, reverse=True
|
|
|
|
)[:5],
|
|
|
|
"tables_count": len(tables),
|
|
|
|
"tables_more": len(tables) > 5,
|
|
|
|
"table_rows_sum": sum((t["count"] or 0) for t in tables.values()),
|
|
|
|
"hidden_table_rows_sum": sum(t["count"] for t in hidden_tables),
|
|
|
|
"hidden_tables_count": len(hidden_tables),
|
|
|
|
"views_count": len(views),
|
|
|
|
}
|
|
|
|
)
|
2018-06-15 06:51:23 +00:00
|
|
|
if as_format:
|
2018-05-13 12:44:22 +00:00
|
|
|
headers = {}
|
|
|
|
if self.ds.cors:
|
2018-05-13 12:55:15 +00:00
|
|
|
headers["Access-Control-Allow-Origin"] = "*"
|
2018-05-13 12:44:22 +00:00
|
|
|
return response.HTTPResponse(
|
2018-05-13 12:55:15 +00:00
|
|
|
json.dumps({db["name"]: db for db in databases}, cls=CustomJSONEncoder),
|
|
|
|
content_type="application/json",
|
2018-05-13 12:44:22 +00:00
|
|
|
headers=headers,
|
|
|
|
)
|
|
|
|
else:
|
|
|
|
return self.render(
|
2018-05-13 12:55:15 +00:00
|
|
|
["index.html"],
|
2018-05-13 12:44:22 +00:00
|
|
|
databases=databases,
|
2018-08-13 14:56:50 +00:00
|
|
|
metadata=self.ds.metadata(),
|
2018-05-13 12:44:22 +00:00
|
|
|
datasette_version=__version__,
|
|
|
|
)
|