import os
import fnmatch

from . import base
from tuned.utils.commands import commands
from tuned import logs

log = logs.get()
cmd = commands()

SYSFS_DIR = "/sys/devices/system/cpu/"

class package2cpus(base.Function):
	"""
	Provides cpu device list for a package (socket)
	"""

	def __init__(self):
		super(package2cpus, self).__init__("package2cpus", 0)

	def execute(self, args):
		if not super(package2cpus, self).execute(args):
			return None

		if len(args) <= 0:
			return None

		try:
			all_cpus = fnmatch.filter(os.listdir(SYSFS_DIR), "cpu[0-9]*")
		except OSError:
			return None

		devices = []

		for cpu in all_cpus:
			f = SYSFS_DIR + cpu + "/topology/physical_package_id"
			if not os.path.exists(f):
				log.warning("File '%s' does not exist" % f)
				continue

			value = cmd.read_file(f)

			try:
				package_id = int(value)
			except ValueError:
				log.warning("Invalid package id '%s' for cpu '%s'" % (value, cpu))
				continue

			for package_pattern in args:
				try:
					this_package_id = int(package_pattern)
				except ValueError:
					if fnmatch.fnmatch(value, package_pattern):
						devices.append(cpu)
				else:
					if package_id == this_package_id:
						devices.append(cpu)

		return ",".join(devices) if len(devices) > 0 else None