Python-сообщество

# -*- coding: utf-8 -*-

"""
Учет внутриофисных расходов
"""
import itertools

class Collaborator:
   """
   Сотрудник
   """
   def __init__(self, code, family, name, patronym):
     self.code = code
     self.family = family
     self.name = name
     self.patronym = patronym

   def __str__(self):
     return "%s %.2s. %.2s." % (self.family, self.name, self.patronym)

   def __hash__(self):
     return hash(self.code)

   def __eq__(self, other):
     return self.code == other.code

class Otdel:
  """
  Отдел
  """
  def __init__(self, name):
    self.name = name
    self.collaborators = set()
    self.otdels = set()

  def __hash__(self):
    return hash(self.name)

  def __eq__(self, other):
    return self.name == other.name

  def __iter__(self):
    i = iter(self.collaborators)
    for s in self.otdels:
      i = itertools.chain(i, iter(s))
    return i

  def add(self, collaborator):
    assert collaborator not in self
    self.collaborators.add(collaborator)

  def addOtdel(self, otdel):
    assert otdel not in self.otdels
    assert not set(self).intersection(set(otdel))
    self.otdels.add(otdel)

  def iterOtdel(self):
    i = iter(self.otdels)
    for s in self.otdels:
      i = itertools.chain(i, s.iterOtdel())
    return i

class RashodType:
  """
  Вид расходов
  """
  def __init__(self, name):
    self.name = name

  def __hash__(self):
    return hash(self.name)

  def __eq__(self, other):
    return self.name == other.name

class RashodTypes(set):
  """
  Виды расходов
  """
  def add(self, rashodType):
    assert rashodType not in self
    set.add(self, rashodType)

class Rashod:
  """
  Расход
  """
  def __init__(self, rashod, rashodType):
    self.sum = rashod
    self.type = rashodType

  def __hash__(self):
    return hash(self.sum)

  def __eq__(self, other):
    return self.sum == other.sum

class Rashods(set):
  """
  Расходы
  """
  def __init__(self, rashodTypes):
    set.__init__(self)
    self.rashodTypes = rashodTypes

  def add(self, rashod):
    assert rashod.type in self.rashodTypes
    set.add(self, rashod)

class RashodRecord:
  """
  Запись в учете внутриофисных расходов
  """
  def __init__(self, rashod, collaborator):
    self.rashod = rashod
    self.collaborator = collaborator

  def __hash__(self):
    return hash((self.rashod, self.collaborator))

  def __eq__(self, other):
    return self.rashod == other.rashod and \
           self.collaborator == other.collaborator

class RashodDir(set):
  """
  Расходный справочник
  """
  def __init__(self, rashods, otdel):
    set.__init__(self)
    self.rashods = rashods
    self.otdel = otdel

  def add(self, rashodRecord):
    assert rashodRecord.collaborator in self.otdel
    assert rashodRecord not in self
    set.add(self, rashodRecord)

if __name__ == '__main__':
  import zzcsv

  rashodDir = zzcsv.load()

  for s in rashodDir.otdel.iterOtdel():
    if s.name == 'отдел кадров':
      for r in rashodDir:
        if r.collaborator in s and r.collaborator.name.find('ан') >= 0:
          print r.rashod.sum, "%s %s. %s."% \
                (r.collaborator.family, r.collaborator.name[:2], r.collaborator.patronym[:2])
      break

  for s in rashodDir.otdel.iterOtdel():
    if s.name == 'руководство':
      for r in rashodDir:
        if r.collaborator in s and r.collaborator.family.find('ск') >= 0:
          print r.rashod.sum, "%s %s. %s."% \
               (r.collaborator.family, r.collaborator.name[:2], r.collaborator.patronym[:2])
      break

# -*- coding: utf-8 -*-

import os, csv

import rashoddir

def save(rashodDir):
  def write(otdel):
    for s in otdel.otdels:
      writer.writerow((s.name, otdel.name))
      write(s)

  writer = csv.writer(open(os.path.join(os.curdir, 'otdel.csv'), 'wb'), delimiter=';')
  writer.writerow((rashodDir.otdel.name, None))
  write(rashodDir.otdel)

  def find(c, otdel):
    if otdel.collaborators:
      return otdel.name
    else:
      for s in otdel.otdels:
        r = find(c, s)

        if r:
          return r

  writer = csv.writer(open(os.path.join(os.curdir, 'mts.csv'), 'wb'), delimiter=';')
  for r in rashodDir:
    writer.writerow((r.rashod.sum, r.collaborator.code, r.collaborator.family, r.collaborator.name, r.collaborator.patronym, find(r.collaborator, rashodDir.otdel), r.rashod.type.name))

def load():
  otdel = {}
  for rec in csv.reader(open(os.path.join(os.curdir, 'otdel.csv'), 'rb'), delimiter=';'):
    otdel[rec[0]] = rashoddir.Otdel(rec[0])
    if rec[1]:
      otdel[rec[1]].addOtdel(otdel[rec[0]])
    else:
      rashodDir = rashoddir.RashodDir(rashoddir.Rashods(rashoddir.RashodTypes()), otdel[rec[0]])

  rashods, rashodTypes, collaborators = {}, {}, {}
  for rec in csv.reader(open(os.path.join(os.curdir, 'mts.csv'), 'rb'), delimiter=';'):
    if rec[6] not in rashodTypes:
      rashodTypes[rec[6]] = rashoddir.RashodType(rec[6])
      rashodDir.rashods.rashodTypes.add(rashodTypes[rec[6]])
    if rec[0] not in rashods:
      rashods[rec[0]] = rashoddir.Rashod(rec[0], rashodTypes[rec[6]])
      rashodDir.rashods.add(rashods[rec[0]])
    key = int(rec[1])
    if key not in collaborators:
      collaborators[key] = rashoddir.Collaborator(key, rec[2], rec[3], rec[4])
      otdel[rec[5]].add(collaborators[key])
    rashodDir.add(rashoddir.RashodRecord(rashods[rec[0]], collaborators[key]))

  return rashodDir

if __name__ == '__main__':
  rashodDir = load()

  for s in rashodDir.otdel.iterOtdel():
    if s.name == 'отдел кадров':
      for r in rashodDir:
       if r.collaborator in s and r.collaborator.name.find('ан') >= 0:
          print r.rashod.sum, "%s %s. %s."% \
                (r.collaborator.family, r.collaborator.name[:2], r.collaborator.patronym[:2])
      break

  for s in rashodDir.otdel.iterOtdel():
    if s.name == 'руководство':
      for r in rashodDir:
        if r.collaborator in s and r.collaborator.family.find('ск') >= 0:
          print r.rashod.sum, "%s %s. %s."% \
                (r.collaborator.family, r.collaborator.name[:2], r.collaborator.patronym[:2])
      break

  save(rashodDir)

МТС;
руководство;МТС
пресс-секретарь;руководство
зам. директор;руководство
директор;руководство
сервис;МТС
уборщик;сервис
дворник;сервис
отдел кадров;МТС
начальник;отдел кадров
сотрудник;отдел кадров

100 руб;4;Глухова;Анна;Юрьевна;начальник;канцтовары
100 руб;7;Звонов;Виталий;Степанович;дворник;канцтовары
200 руб;5;Цветков;Иван;Викторович;сотрудник;связь
35 руб;6;Маринин;Игорь;Олегович;уборщик;хоз. нужды
500 руб;3;Лаптева;Иврина;Сергеевна;пресс-секретарь;канцтовары
2 000 руб;1;Иванов;Иван;Андреевич;директор;связь
5 000 руб;2;Петров;Максим;Павлович;зам. директорь;хоз. нужды

if key not in collaborators:
      collaborators[key] = rashoddir.Collaborator(key, rec[2], rec[3], rec[4])
      otdel[rec[5]].add(collaborators[key])
    rashodDir.add(rashoddir.RashodRecord(rashods[rec[0]], collaborators[key]))