# vi: ts=4 expandtab
#
#    Copyright (C) 2016 Alibaba Cloud Inc.
#
#    Author: jingni.wjj <jingni.wjj@alibaba-inc.com>

import os
import re

from cloudinit.distros import rhel, get_lsb_release
from cloudinit import log as logging
from cloudinit import util
from cloudinit.distros import rhel_util

LOG = logging.getLogger(__name__)


class Distro(rhel.Distro):

    def __init__(self, name, cfg, path):
        rhel.Distro.__init__(self, name, cfg, path)
        self.cname = 'Anolis'
        self.rpmgpgkey_path = None

    def get_fn_release(self):
        reg = re.compile('Anolis OS release (\d+.\d+).*')
        check_files = ('/etc/issue',
                       '/etc/issue.net',
                       '/etc/system-release',
                       '/etc/anolis-release',
                       '/etc/redhat-release')
        for fn in check_files:
            if os.path.exists(fn):
                out = reg.findall(util.load_file(fn).strip())
                if out:
                    return out[0]
        return ''