From a1c937350ee5df35cbba102fefa28077bf17f407 Mon Sep 17 00:00:00 2001 From: Joey Hines Date: Thu, 13 Sep 2018 12:08:36 -0500 Subject: [PATCH] Tunnel color changed to tunnel direction. --- geoffrey/DatabaseModels.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/geoffrey/DatabaseModels.py b/geoffrey/DatabaseModels.py index df25641..406da9a 100644 --- a/geoffrey/DatabaseModels.py +++ b/geoffrey/DatabaseModels.py @@ -151,11 +151,11 @@ class Tunnel(SQL_Base): location_id = Column(Integer, ForeignKey('geoffrey_locations.id', ondelete='CASCADE')) location = relationship("Location", back_populates="tunnel", lazy="joined") - def __init__(self, owner, tunnel_color, tunnel_number, location=None): + def __init__(self, owner, tunnel_direction, tunnel_number, location=None): try: self.owner = owner self.location = location - self.tunnel_direction = TunnelDirection.str_to_tunnel_dir(tunnel_color) + self.tunnel_direction = TunnelDirection.str_to_tunnel_dir(tunnel_direction) self.tunnel_number = tunnel_number except (ValueError, IndexError): raise TunnelInitError