Add missing enum imports (#157)

This commit is contained in:
cclauss 2018-05-07 03:26:01 +02:00 committed by Donne Martin
parent 4a7d2e4525
commit a36048346e
2 changed files with 6 additions and 2 deletions

View File

@ -1,4 +1,5 @@
from abc import ABCMeta from abc import ABCMeta
from enum import Enum
class UserService(object): class UserService(object):

View File

@ -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