Add missing enum imports (#157)
parent
4a7d2e4525
commit
a36048346e
|
@ -1,4 +1,5 @@
|
||||||
from abc import ABCMeta
|
from abc import ABCMeta
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
|
||||||
class UserService(object):
|
class UserService(object):
|
||||||
|
|
|
@ -1,12 +1,16 @@
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
|
||||||
class DefaultCategories(Enum):
|
class DefaultCategories(Enum):
|
||||||
|
|
||||||
HOUSING = 0
|
HOUSING = 0
|
||||||
FOOD = 1
|
FOOD = 1
|
||||||
GAS = 2
|
GAS = 2
|
||||||
SHOPPING = 3
|
SHOPPING = 3
|
||||||
...
|
# ...
|
||||||
|
|
||||||
|
|
||||||
seller_category_map = {}
|
seller_category_map = {}
|
||||||
seller_category_map['Exxon'] = DefaultCategories.GAS
|
seller_category_map['Exxon'] = DefaultCategories.GAS
|
||||||
|
@ -44,4 +48,3 @@ class Budget(object):
|
||||||
|
|
||||||
def override_category_budget(self, category, amount):
|
def override_category_budget(self, category, amount):
|
||||||
self.categories_to_budget_map[category] = amount
|
self.categories_to_budget_map[category] = amount
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue