from typing import List

from meta import Meta

from .partition_plan import PartitionPlan


class DatabasePartitionPlan(PartitionPlan):
    def __init__(self, meta: Meta):
        self.meta = meta

    def groups(self) -> List[List[str]]:
        groups = []
        for database in self.meta.databases():
            g = []
            for table in database.tables:
                table_id = f"{database.name}.{table.name}"
                g.append(table_id)
            groups.append(g)
        return groups