ip_type_grains.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from salt.modules import network


def ip_type():
    grains = {}
    ip_addresses = network.ipaddrs()
    if ip_addresses:
        for ip in ip_addresses:
            if network.is_private(ip):
                grains['local_ip'] = ip
            else:
                grains['public_ip'] = ip
    return grains

if __name__ == "__main__":
    print ip_type()