init

app/util/visa_availability_scraper_playwright.py

@@ -0,0 +1,394 @@
+import asyncio
+import json
+from typing import Dict, Optional, List
+from playwright.async_api import async_playwright
+
+class PassportIndexVisaScraper:
+    def __init__(self, debug: bool = True):
+        """
+        Initialize the Passport Index visa scraper using Playwright
+        
+        Args:
+            debug: Enable debug output
+        """
+        self.base_url = "https://www.passportindex.org/travel-visa-checker/"
+        self.api_url = "https://www.passportindex.org/core/visachecker.php"
+        self.debug = debug
+        self.browser = None
+        self.context = None
+        self.page = None
+        
+    async def __aenter__(self):
+        """Initialize browser with stealth mode"""
+        self.playwright = await async_playwright().start()
+        
+        self.browser = await self.playwright.chromium.launch(
+            headless=True,  # Using headless mode
+            args=[
+                '--disable-blink-features=AutomationControlled',
+                '--disable-dev-shm-usage',
+                '--no-sandbox',
+                '--disable-setuid-sandbox',
+                '--disable-web-security',
+                '--disable-features=IsolateOrigins,site-per-process'
+            ]
+        )
+        
+        # Create context with realistic settings
+        self.context = await self.browser.new_context(
+            viewport={'width': 1920, 'height': 1080},
+            user_agent='Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/140.0.0.0 Safari/537.36',
+            locale='en-US',
+            timezone_id='America/New_York'
+        )
+        
+        self.page = await self.context.new_page()
+
+        await self.page.add_init_script("""
+            // Override the navigator.webdriver property
+            Object.defineProperty(navigator, 'webdriver', {
+                get: () => undefined
+            });
+
+            // Override chrome property
+            window.chrome = {
+                runtime: {}
+            };
+
+            // Override permissions
+            const originalQuery = window.navigator.permissions.query;
+            window.navigator.permissions.query = (parameters) => (
+                parameters.name === 'notifications' ?
+                    Promise.resolve({ state: Notification.permission }) :
+                    originalQuery(parameters)
+            );
+        """)
+        
+        if self.debug:
+            print("🚀 Browser initialized with stealth mode")
+        
+        return self
+    
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        """Clean up browser resources"""
+        if self.page:
+            await self.page.close()
+        if self.context:
+            await self.context.close()
+        if self.browser:
+            await self.browser.close()
+        if self.playwright:
+            await self.playwright.stop()
+        
+        if self.debug:
+            print("🔒 Browser closed")
+    
+    async def initialize_session(self) -> bool:
+        """
+        Navigate to the website and wait for it to load properly
+        """
+        try:
+            if self.debug:
+                print("📱 Initializing session...")
+            
+            # Navigate to the page
+            try:
+                response = await self.page.goto(
+                    self.base_url,
+                    wait_until='domcontentloaded',
+                    timeout=30000
+                )
+                await self.page.wait_for_timeout(3000)
+                
+                # Get the cl value from the page
+                cl_value = await self.page.evaluate("""
+                    () => {
+                        const clInput = document.querySelector('#cl');
+                        return clInput ? clInput.value : 'bc2140a2d83928ce1112d01e610bad89';
+                    }
+                """)
+                
+                if self.debug:
+                    print(f"✅ Page loaded, session ID: {cl_value}")
+                    
+                return True
+                
+            except Exception as e:
+                if self.debug:
+                    print(f"⚠️ Page load issue: {e}, continuing anyway...")
+                return True
+                
+        except Exception as e:
+            print(f"❌ Error initializing session: {e}")
+            return False
+    
+    async def check_visa_requirement_browser(self, passport_country: str, destination_country: str) -> Optional[Dict]:
+        """
+        Check visa requirements using browser automation
+        
+        Args:
+            passport_country: Two-letter country code for passport
+            destination_country: Two-letter country code for destination
+        
+        Returns:
+            Dictionary with visa information or None if failed
+        """
+        try:
+            if self.debug:
+                print(f"🌐 Checking {passport_country.upper()} → {destination_country.upper()}")
+            
+            # Get the current session ID from the page
+            cl_value = await self.page.evaluate("""
+                () => {
+                    const clInput = document.querySelector('#cl');
+                    return clInput ? clInput.value : 'bc2140a2d83928ce1112d01e610bad89';
+                }
+            """)
+            
+            # Make the API request through the browser with proper argument passing
+            result = await self.page.evaluate("""
+                async (args) => {
+                    const [passport, destination, sessionId] = args;
+                    const formData = new URLSearchParams();
+                    formData.append('d', destination);
+                    formData.append('s', passport);
+                    formData.append('cl', sessionId);
+                    
+                    try {
+                        const response = await fetch('https://www.passportindex.org/core/visachecker.php', {
+                            method: 'POST',
+                            headers: {
+                                'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
+                                'X-Requested-With': 'XMLHttpRequest',
+                                'Accept': 'application/json, text/javascript, */*; q=0.01'
+                            },
+                            body: formData.toString(),
+                            credentials: 'include'
+                        });
+                        
+                        if (!response.ok) {
+                            throw new Error(`HTTP ${response.status}`);
+                        }
+                        
+                        const data = await response.json();
+                        return data;
+                    } catch (error) {
+                        return { error: error.message };
+                    }
+                }
+            """, [passport_country.lower(), destination_country.lower(), cl_value])
+            
+            if result and 'error' not in result:
+                if self.debug:
+                    print(f"✅ Got result: {result}")
+                return result
+            elif result and 'error' in result:
+                print(f"❌ API Error: {result['error']}")
+                return None
+            else:
+                return None
+                
+        except Exception as e:
+            print(f"❌ Error checking visa requirement: {e}")
+            return None
+    
+    async def check_visa_interactive(self, passport_country: str, destination_country: str) -> Optional[Dict]:
+        """
+        Alternative method: Use the interactive UI to check visa requirements
+        """
+        try:
+            if self.debug:
+                print(f"🖱️ Using interactive method for {passport_country.upper()} → {destination_country.upper()}")
+            
+            # Click on the passport selector
+            await self.page.click('.vch-select-pass')
+            await self.page.wait_for_timeout(500)
+            
+            # Find and click the country in the list
+            passport_selector = f'.vch-passports .s-div[data-ccode="{passport_country.lower()}"]'
+            await self.page.wait_for_selector(passport_selector, timeout=5000)
+            await self.page.click(passport_selector)
+            await self.page.wait_for_timeout(500)
+            
+            # Click on the destination selector
+            await self.page.click('.vch-select-des')
+            await self.page.wait_for_timeout(500)
+            
+            # Find and click the destination country
+            dest_selector = f'.vch-destinations .s-div[data-ccode="{destination_country.lower()}"]'
+            await self.page.wait_for_selector(dest_selector, timeout=5000)
+            await self.page.click(dest_selector)
+            await self.page.wait_for_timeout(1000)
+            
+            # Get the result from the page
+            result = await self.page.evaluate("""
+                () => {
+                    const resultElement = document.querySelector('.vch-result');
+                    if (resultElement) {
+                        const text = resultElement.querySelector('.text');
+                        const days = resultElement.querySelector('.days');
+                        return {
+                            text: text ? text.textContent : '',
+                            days: days ? days.textContent : '',
+                            pass: '""" + passport_country.lower() + """',
+                            dest: '""" + destination_country.upper() + """'
+                        };
+                    }
+                    return null;
+                }
+            """)
+            
+            return result
+            
+        except Exception as e:
+            if self.debug:
+                print(f"❌ Interactive method failed: {e}")
+            return None
+    
+    async def check_multiple_destinations(self, passport_country: str, destinations: List[str], delay: float = 2.0) -> Dict:
+        """
+        Check visa requirements for multiple destinations
+        
+        Args:
+            passport_country: Two-letter country code for passport
+            destinations: List of two-letter country codes for destinations
+            delay: Delay between requests in seconds
+        
+        Returns:
+            Dictionary mapping destination codes to visa information
+        """
+        results = {}
+        
+        for i, dest in enumerate(destinations, 1):
+            print(f"\n[{i}/{len(destinations)}] Checking {passport_country.upper()} → {dest.upper()}...")
+            
+            # Try API method first
+            result = await self.check_visa_requirement_browser(passport_country, dest)
+            
+            # If API fails, try interactive method
+            if not result:
+                result = await self.check_visa_interactive(passport_country, dest)
+            
+            if result:
+                results[dest] = result
+                text = result.get('text', 'No text available')
+                print(f"  ✅ Result: {text}")
+            else:
+                results[dest] = None
+                print(f"  ❌ Failed to get result")
+            
+            # Rate limiting
+            if i < len(destinations):
+                print(f"  ⏳ Waiting {delay} seconds...")
+                await asyncio.sleep(delay)
+        
+        return results
+    
+    def format_result(self, result: Dict) -> str:
+        """Format a single result for display"""
+        if not result:
+            return "No information available"
+        
+        text = result.get('text', 'N/A')
+        dest = result.get('dest', 'N/A')
+        passport = result.get('pass', 'N/A')
+        
+        return f"{passport.upper()} → {dest.upper()}: {text}"
+
+
+async def main():
+    """Main function to demonstrate usage"""
+    print("="*60)
+    print(" Passport Index Visa Checker (Playwright)")
+    print("="*60)
+    
+    async with PassportIndexVisaScraper(debug=True) as scraper:
+        # Initialize session
+        if not await scraper.initialize_session():
+            print("❌ Failed to initialize session")
+            return
+        
+        print("\n" + "="*60)
+        print(" Testing visa requirements...")
+        print("="*60)
+        
+        # Test single visa requirement
+        print("\n📍 Single visa check: US → GB")
+        print("-" * 40)
+        result = await scraper.check_visa_requirement_browser('us', 'gb')
+        if result:
+            print(f"Result: {scraper.format_result(result)}")
+        else:
+            print("Trying interactive method...")
+            result = await scraper.check_visa_interactive('us', 'gb')
+            if result:
+                print(f"Result: {scraper.format_result(result)}")
+        
+        # Test multiple destinations
+        print("\n📍 Multiple destinations for US passport:")
+        print("-" * 40)
+        destinations = ['ca', 'mx', 'jp', 'au']  # Canada, Mexico, Japan, Australia
+        results = await scraper.check_multiple_destinations('us', destinations, delay=2.0)
+        
+        print("\n📊 Summary:")
+        for dest, result in results.items():
+            if result:
+                print(f"  ✅ {scraper.format_result(result)}")
+            else:
+                print(f"  ❌ US → {dest.upper()}: Failed")
+
+
+COUNTRY_CODES = {
+    'af': 'Afghanistan', 'al': 'Albania', 'dz': 'Algeria', 'ad': 'Andorra',
+    'ao': 'Angola', 'ag': 'Antigua and Barbuda', 'ar': 'Argentina', 'am': 'Armenia',
+    'au': 'Australia', 'at': 'Austria', 'az': 'Azerbaijan', 'bs': 'Bahamas',
+    'bh': 'Bahrain', 'bd': 'Bangladesh', 'bb': 'Barbados', 'by': 'Belarus',
+    'be': 'Belgium', 'bz': 'Belize', 'bj': 'Benin', 'bt': 'Bhutan',
+    'bo': 'Bolivia', 'ba': 'Bosnia and Herzegovina', 'bw': 'Botswana', 'br': 'Brazil',
+    'bn': 'Brunei', 'bg': 'Bulgaria', 'bf': 'Burkina Faso', 'bi': 'Burundi',
+    'kh': 'Cambodia', 'cm': 'Cameroon', 'ca': 'Canada', 'cv': 'Cape Verde',
+    'cf': 'Central African Republic', 'td': 'Chad', 'cl': 'Chile', 'cn': 'China',
+    'co': 'Colombia', 'km': 'Comoros', 'cg': 'Congo', 'cr': 'Costa Rica',
+    'hr': 'Croatia', 'cu': 'Cuba', 'cy': 'Cyprus', 'cz': 'Czech Republic',
+    'dk': 'Denmark', 'dj': 'Djibouti', 'dm': 'Dominica', 'do': 'Dominican Republic',
+    'ec': 'Ecuador', 'eg': 'Egypt', 'sv': 'El Salvador', 'gq': 'Equatorial Guinea',
+    'er': 'Eritrea', 'ee': 'Estonia', 'et': 'Ethiopia', 'fj': 'Fiji',
+    'fi': 'Finland', 'fr': 'France', 'ga': 'Gabon', 'gm': 'Gambia',
+    'ge': 'Georgia', 'de': 'Germany', 'gh': 'Ghana', 'gr': 'Greece',
+    'gd': 'Grenada', 'gt': 'Guatemala', 'gn': 'Guinea', 'gw': 'Guinea-Bissau',
+    'gy': 'Guyana', 'ht': 'Haiti', 'hn': 'Honduras', 'hu': 'Hungary',
+    'is': 'Iceland', 'in': 'India', 'id': 'Indonesia', 'ir': 'Iran',
+    'iq': 'Iraq', 'ie': 'Ireland', 'il': 'Israel', 'it': 'Italy',
+    'jm': 'Jamaica', 'jp': 'Japan', 'jo': 'Jordan', 'kz': 'Kazakhstan',
+    'ke': 'Kenya', 'ki': 'Kiribati', 'kp': 'North Korea', 'kr': 'South Korea',
+    'kw': 'Kuwait', 'kg': 'Kyrgyzstan', 'la': 'Laos', 'lv': 'Latvia',
+    'lb': 'Lebanon', 'ls': 'Lesotho', 'lr': 'Liberia', 'ly': 'Libya',
+    'li': 'Liechtenstein', 'lt': 'Lithuania', 'lu': 'Luxembourg', 'mk': 'Macedonia',
+    'mg': 'Madagascar', 'mw': 'Malawi', 'my': 'Malaysia', 'mv': 'Maldives',
+    'ml': 'Mali', 'mt': 'Malta', 'mh': 'Marshall Islands', 'mr': 'Mauritania',
+    'mu': 'Mauritius', 'mx': 'Mexico', 'fm': 'Micronesia', 'md': 'Moldova',
+    'mc': 'Monaco', 'mn': 'Mongolia', 'me': 'Montenegro', 'ma': 'Morocco',
+    'mz': 'Mozambique', 'mm': 'Myanmar', 'na': 'Namibia', 'nr': 'Nauru',
+    'np': 'Nepal', 'nl': 'Netherlands', 'nz': 'New Zealand', 'ni': 'Nicaragua',
+    'ne': 'Niger', 'ng': 'Nigeria', 'no': 'Norway', 'om': 'Oman',
+    'pk': 'Pakistan', 'pw': 'Palau', 'pa': 'Panama', 'pg': 'Papua New Guinea',
+    'py': 'Paraguay', 'pe': 'Peru', 'ph': 'Philippines', 'pl': 'Poland',
+    'pt': 'Portugal', 'qa': 'Qatar', 'ro': 'Romania', 'ru': 'Russia',
+    'rw': 'Rwanda', 'kn': 'Saint Kitts and Nevis', 'lc': 'Saint Lucia',
+    'vc': 'Saint Vincent and the Grenadines', 'ws': 'Samoa', 'sm': 'San Marino',
+    'st': 'Sao Tome and Principe', 'sa': 'Saudi Arabia', 'sn': 'Senegal',
+    'rs': 'Serbia', 'sc': 'Seychelles', 'sl': 'Sierra Leone', 'sg': 'Singapore',
+    'sk': 'Slovakia', 'si': 'Slovenia', 'sb': 'Solomon Islands', 'so': 'Somalia',
+    'za': 'South Africa', 'es': 'Spain', 'lk': 'Sri Lanka', 'sd': 'Sudan',
+    'sr': 'Suriname', 'sz': 'Swaziland', 'se': 'Sweden', 'ch': 'Switzerland',
+    'sy': 'Syria', 'tw': 'Taiwan', 'tj': 'Tajikistan', 'tz': 'Tanzania',
+    'th': 'Thailand', 'tl': 'Timor-Leste', 'tg': 'Togo', 'to': 'Tonga',
+    'tt': 'Trinidad and Tobago', 'tn': 'Tunisia', 'tr': 'Turkey', 'tm': 'Turkmenistan',
+    'tv': 'Tuvalu', 'ug': 'Uganda', 'ua': 'Ukraine', 'ae': 'United Arab Emirates',
+    'gb': 'United Kingdom', 'us': 'United States', 'uy': 'Uruguay', 'uz': 'Uzbekistan',
+    'vu': 'Vanuatu', 've': 'Venezuela', 'vn': 'Vietnam', 'ye': 'Yemen',
+    'zm': 'Zambia', 'zw': 'Zimbabwe'
+}
+
+if __name__ == "__main__":
+    asyncio.run(main())