From 95824fd246f82dc29a9380457b95b53b7ca3a3a5 Mon Sep 17 00:00:00 2001
From: Patrick Arminio <patrick.arminio@gmail.com>
Date: Wed, 24 Nov 2021 15:12:00 +0000
Subject: [PATCH] Get poetry cache implementation when requested

---
 src/cache-distributions/cache-factory.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/cache-distributions/cache-factory.ts b/src/cache-distributions/cache-factory.ts
index 3e363a0..7becf53 100644
--- a/src/cache-distributions/cache-factory.ts
+++ b/src/cache-distributions/cache-factory.ts
@@ -1,9 +1,11 @@
 import PipCache from './pip-cache';
 import PipenvCache from './pipenv-cache';
+import PoetryCache from './poetry-cache';
 
 export enum PackageManagers {
   Pip = 'pip',
-  Pipenv = 'pipenv'
+  Pipenv = 'pipenv',
+  Poetry = 'poetry'
 }
 
 export function getCacheDistributor(
@@ -16,6 +18,8 @@ export function getCacheDistributor(
       return new PipCache(pythonVersion, cacheDependencyPath);
     case PackageManagers.Pipenv:
       return new PipenvCache(pythonVersion, cacheDependencyPath);
+    case PackageManagers.Poetry:
+      return new PoetryCache(pythonVersion, cacheDependencyPath);
     default:
       throw new Error(`Caching for '${packageManager}' is not supported`);
   }