Python使用cusom类进入Main

时间:2018-10-12 15:09:46

标签: python

我已经开发了一个自定义的python脚本。我正在使用main运行所有脚本功能。这是一个结构示例:

import ssl
import itertools
import decimal


def main():

if __name__ == '__main__':
main()

现在我有一个需要在main中使用的具有不同def的特定类。 我在互联网上看到的这个Class示例:

class GeoJSON:
"""GeoJSON class which allows to calculate bbox"""
def __init__(self, geojson):
    if geojson['type'] == 'FeatureCollection':
        self.coords = list(self._flatten([f['geometry']['coordinates']
                       for f in geojson['features']]))
        self.features_count = len(geojson['features'])
    elif geojson['type'] == 'Feature':
        self.coords = list(self._flatten([
                    geojson['geometry']['coordinates']]))
        self.features_count = 1
    else:
        self.coords = list(self._flatten([geojson['coordinates']]))
        self.features_count = 1

def _flatten(self, l):
    for val in l:
        if isinstance(val, list):
            for subval in self._flatten(val):
                yield subval
        else:
            yield val

def bbox(self):
    return [min(self.coords[::2]), min(self.coords[1::2]),
            max(self.coords[::2]), max(self.coords[1::2])]

现在我可以使用GeoJson类并将其定义为我的主要对象了吗?

谢谢

1 个答案:

答案 0 :(得分:2)

您的main函数不是神奇的,您可以在此处实例化类,就像在其他任何地方一样。

class A():
    pass

def main():
    a = A()  # a is now an instance of A, you can create multiple instances
    b = A()
    a_third_instance = A()