DIRECTORY COMPARISON

#!/usr/bin/env python3
"""DIRECTORY COMPARISON"""
import os, tempfile
temp = tempfile.gettempdir()
dir1 = os.path.join(temp, "dir1")
dir2 = os.path.join(temp, "dir2")
os.makedirs(dir1)
os.makedirs(dir2)
open(os.path.join(dir1, "both.txt"), 'w').write("test")
open(os.path.join(dir1, "only1.txt"), 'w').write("test")
open(os.path.join(dir2, "both.txt"), 'w').write("test")
open(os.path.join(dir2, "only2.txt"), 'w').write("test")
files1 = set(os.listdir(dir1))
files2 = set(os.listdir(dir2))
print(f"Only in dir1: {files1 - files2}")
print(f"Only in dir2: {files2 - files1}")
print(f"In both: {files1 & files2}")
import shutil
shutil.rmtree(dir1)
shutil.rmtree(dir2)