Added user type attribute

2017-08-25 14:59:01 +01:00 · 2017-08-25 14:59:01 +01:00 · 0a9ca57689
parent e886eb2ce6
commit 0a9ca57689
1 changed files with 10 additions and 1 deletions
--- a/lib/Pear/LocalLoop/Schema/Result/User.pm
+++ b/lib/Pear/LocalLoop/Schema/Result/User.pm
@ -149,8 +149,17 @@ sub name {
  } elsif ( defined $self->organisation_id ) {
    return $self->organisation->name;
  } else {
-    return undef;
+    return;
  }
 }

+sub type {
+  my $self = shift;
+
+  if ( defined $self->customer_id ) {
+    return "customer";
+  }
+  return "organisation";
+}
+
 1;