#!/usr/bin/env python
# -*- coding: utf-8 -*-
import zipfile
import rarfile
from os import listdir
from os.path import isfile, join
filePathToZips = "C:\\Users\\OscarAndersson\\Desktop\\script\\"

def checkRar(file, filePath):
  rf = rarfile.RarFile(filePath)
  for rarItem in rf.infolist():
    checkFileEndingIsObj(rarItem.filename, filePath)

def checkZip(file, filePath):
  zipArchive = zipfile.ZipFile(filePath, 'r')
  for zipItem in zipArchive.infolist():
    checkFileEndingIsObj(zipItem.filename, filePath)

def checkFileEndingIsObj(fileName, filePath):
  if fileName.endswith('.OBJ') or fileName.endswith('.obj'):
    try:
      completePath = join(filePathToZips, fileName)
      open(completePath, "x")
    except FileExistsError:
      print('File already exists')


onlyfiles = [f for f in listdir(filePathToZips) if isfile(join(filePathToZips, f))]
for file in onlyfiles:
  filePath = filePathToZips + file
  if file.endswith('.zip'):
    checkZip(file, filePath)
  elif file.endswith('.rar'):
    checkRar(file, filePath)