diff options
Diffstat (limited to 'lib/d2/app/adapters/plot.py')
-rw-r--r-- | lib/d2/app/adapters/plot.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/lib/d2/app/adapters/plot.py b/lib/d2/app/adapters/plot.py index 451f215..ec6075d 100644 --- a/lib/d2/app/adapters/plot.py +++ b/lib/d2/app/adapters/plot.py | |||
@@ -1,7 +1,7 @@ | |||
1 | from d2.app.model.static import StaticData | 1 | from d2.app.model.static import StaticData |
2 | from d2.app.adapters import BaseAdapter | 2 | from d2.app.adapters import BaseAdapter |
3 | from d2.config import Config | 3 | from d2.config import Config |
4 | from d2.db import Plot | 4 | from d2.db import Plot |
5 | 5 | ||
6 | class PlotAdapter(BaseAdapter): | 6 | class PlotAdapter(BaseAdapter): |
7 | 7 | ||
@@ -9,7 +9,7 @@ class PlotAdapter(BaseAdapter): | |||
9 | self._db = db | 9 | self._db = db |
10 | self._log = log | 10 | self._log = log |
11 | self._static = static | 11 | self._static = static |
12 | self._plot = plot | 12 | self._plot = plot |
13 | 13 | ||
14 | @classmethod | 14 | @classmethod |
15 | def load(cls, config=None, static=None): | 15 | def load(cls, config=None, static=None): |
@@ -20,3 +20,8 @@ class PlotAdapter(BaseAdapter): | |||
20 | def get_all(self, plot_ids): | 20 | def get_all(self, plot_ids): |
21 | return self._db.session.query(self._plot).filter( | 21 | return self._db.session.query(self._plot).filter( |
22 | self._plot.id.in_(plot_ids)).all() | 22 | self._plot.id.in_(plot_ids)).all() |
23 | |||
24 | |||
25 | def get_all_from_forges(self, forges): | ||
26 | hold = [forge.plot_id for forge in forges] | ||
27 | return self.get_all(hold) | ||